
    `i                         d dl Zd dlZd dlmZ d dlmZmZmZm	Z	 d dl
mZmZ d Zd Z G d de          Zed	k    r ej                     dS dS )
    N)cuda)float32float64int32void)unittestCUDATestCasec                 @    t          j        |          \  | d<   |d<   d S Nr   )mathfrexp)aryxaryexpargs      |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_frexp_ldexp.pysimple_frexpr      s     CDGVAYYY    c                 6    t          j        ||          | d<   d S r   )r   ldexp)r   r   exps      r   simple_ldexpr      s    jc""DGGGr   c                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )TestCudaFrexpLdexpc           	      F    t          j        t          |d d          t          d d          |                    t                    }d}t          j        d|          }t          j        dt
          j                  } |d         |||           t
          j                            | |d                     | 	                    |d           t
          j
        } |d         |||           t
          j                            | |t
          j
                             | 	                    |d           t
          j        } |d         |||           t
          j                            | |t
          j                             | 	                    |d           d S )No!	@   dtyper   r   o!?   r   )r   jitr   r   r   npzerostestingassert_array_equalassertEqualinfnan)selfnptypenbtypecompiledr   r   r   s          r   template_test_frexpz&TestCudaFrexpLdexp.template_test_frexp   sj   >48DE!!!Hf==>>|LLx(((!28,,,tVS)))

%%dFF8,<,<===###ftVS)))

%%dFF26NN;;;###ftVS)))

%%dFF26NN;;;#####r   c                 v    t          j        t          |d d          |t                              t                    }d}d}t          j        d|          } |d         |||           t
          j                            | |d                     t
          j	        } |d         |||           t
          j                            | |t
          j	                             t
          j
        } |d         |||           t
          j                            | |t
          j
                             d S )Nr    r!   r   r   r   r   )r   r"   r   r   r   r#   r$   r%   r&   r(   r)   )r*   r+   r,   r-   r   r   r   s          r   template_test_ldexpz&TestCudaFrexpLdexp.template_test_ldexp$   s   ;48DFE::;;LIIx(((tS#&&&

%%dFF6NN;;;ftS#&&&

%%dFF26NN;;;ftS#&&&

%%dFF26NN;;;;;r   c                 P    |                      t          j        t                     d S N)r.   r#   r   r*   s    r   test_frexp_f4z TestCudaFrexpLdexp.test_frexp_f44          W55555r   c                 P    |                      t          j        t                     d S r2   )r0   r#   r   r3   s    r   test_ldexp_f4z TestCudaFrexpLdexp.test_ldexp_f47   r5   r   c                 P    |                      t          j        t                     d S r2   )r.   r#   r   r3   s    r   test_frexp_f8z TestCudaFrexpLdexp.test_frexp_f8:   r5   r   c                 P    |                      t          j        t                     d S r2   )r0   r#   r   r3   s    r   test_ldexp_f8z TestCudaFrexpLdexp.test_ldexp_f8=   r5   r   N)	__name__
__module____qualname__r.   r0   r4   r7   r9   r;    r   r   r   r      sn        $ $ $&< < < 6 6 66 6 66 6 66 6 6 6 6r   r   __main__)numpyr#   r   numbar   numba.typesr   r   r   r   numba.cuda.testingr   r	   r   r   r   r<   mainr?   r   r   <module>rF      s               5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5) ) )# # #.6 .6 .6 .6 .6 .6 .6 .6b zHMOOOOO r   