
    `i                     ~    d dl Zd dlmZ d dlmZ d dlZd Z G d de          Ze	dk    r ej
                     dS dS )    N)cuda)CUDATestCasec                 j    | ||                              t          j                  d         }||d<   d S )Nr   )viewnpint32)byte_arrstartstopoutputvals        ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_array_methods.pyreinterpret_array_typer      s2    
5:

#
#BH
-
-a
0CF1III    c                       e Zd Zd ZdS )TestCudaArrayMethodsc                    t           }t          j        |          }t          j        dt          j                  }t          j        t          j                  j        }t          dd|          D ]}||z   }|||         
                    t          j                  d         }t          j        dt          j                  } |d         ||||           |d         }	|                     ||	           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   gots
             r   test_reinterpret_array_typez0TestCudaArrayMethods.test_reinterpret_array_type   s     (&!!9S1118BH%%.1c8,, 	* 	*E8#DeDj)..rx88;FXarx000FF4L5$777)CVS))))	* 	*r   N)__name__
__module____qualname__r#    r   r   r   r      s#        * * * * *r   r   __main__)numpyr   numbar   numba.cuda.testingr   unittestr   r   r$   mainr'   r   r   <module>r.      s              + + + + + +   * * * * *< * * ** zHMOOOOO r   