o
    ۾i                     @   sX   d dl Zd dlmZ d dlmZ d dlZdd ZG dd deZe	dkr*e
  dS dS )	    N)cuda)CUDATestCasec                 C   s$   | ||  tjd }||d< d S )Nr   )viewnpint32)byte_arrstartstopoutputval r   ^/home/ubuntu/.local/lib/python3.10/site-packages/numba/cuda/tests/cudapy/test_array_methods.pyreinterpret_array_type   s   r   c                   @   s   e Zd Zdd ZdS )TestCudaArrayMethodsc           
      C   s   t }t|}tjdtjd}ttjj}t	dd|D ]-}|| }||| 
tjd }tjdtjd}|d |||| |d }	| ||	 qdS )z=
        Reinterpret byte array as int32 in the GPU.
           )dtyper      )r   r   N)r   r   jitr   arangeuint8r   r   itemsizeranger   zerosassertEqual)
selfpyfunckernelr   r   r   r	   expectr
   gotr   r   r   test_reinterpret_array_type   s   
z0TestCudaArrayMethods.test_reinterpret_array_typeN)__name__
__module____qualname__r   r   r   r   r   r      s    r   __main__)numpyr   numbar   numba.cuda.testingr   unittestr   r   r    mainr   r   r   r   <module>   s    