
    sh                         S SK rS SKrS SKJr  S SKJrJrJrJ	r	  S SK
JrJr  S rS r " S S\5      r\S	:X  a  \R"                  " 5         gg)
    N)cuda)float32float64int32void)unittestCUDATestCasec                 B    [         R                  " U5      u  U S'   US'   g Nr   )mathfrexp)aryxaryexpargs      |/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudapy/test_frexp_ldexp.pysimple_frexpr      s    CDGVAY    c                 6    [         R                  " X5      U S'   g r   )r   ldexp)r   r   exps      r   simple_ldexpr      s    jj"DGr   c                   8    \ rS rSrS rS rS rS rS rS r	Sr
g	)
TestCudaFrexpLdexp   c                    [         R                  " [        US S  [        S S  U5      5      " [        5      nSn[
        R                  " SUS9n[
        R                  " S[
        R                  S9nUS   " XVU5        [
        R                  R                  XQ" S5      5        U R                  US5        [
        R                  nUS   " XVU5        [
        R                  R                  XQ" [
        R                  5      5        U R                  US5        [
        R                  nUS   " XVU5        [
        R                  R                  XQ" [
        R                  5      5        U R                  US5        g )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_frexp&TestCudaFrexpLdexp.template_test_frexp   s   88DE!Hf=>|Lxx(!288,tS)


%%dF8,<=#fftS)


%%dF266N;#fftS)


%%dF266N;#r   c                 8   [         R                  " [        US S  U[        5      5      " [        5      nSnSn[
        R                  " SUS9nUS   " XdU5        [
        R                  R                  Xa" S5      5        [
        R                  nUS   " XdU5        [
        R                  R                  Xa" [
        R                  5      5        [
        R                  nUS   " XdU5        [
        R                  R                  Xa" [
        R                  5      5        g )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_ldexp&TestCudaFrexpLdexp.template_test_ldexp$   s    88DFE:;LIxx(t#&


%%dF6N;fft#&


%%dF266N;fft#&


%%dF266N;r   c                 L    U R                  [        R                  [        5        g N)r/   r$   r   r+   s    r   test_frexp_f4 TestCudaFrexpLdexp.test_frexp_f44         W5r   c                 L    U R                  [        R                  [        5        g r5   )r2   r$   r   r6   s    r   test_ldexp_f4 TestCudaFrexpLdexp.test_ldexp_f47   r9   r   c                 L    U R                  [        R                  [        5        g r5   )r/   r$   r   r6   s    r   test_frexp_f8 TestCudaFrexpLdexp.test_frexp_f8:   r9   r   c                 L    U R                  [        R                  [        5        g r5   )r2   r$   r   r6   s    r   test_ldexp_f8 TestCudaFrexpLdexp.test_ldexp_f8=   r9   r    N)__name__
__module____qualname____firstlineno__r/   r2   r7   r;   r>   rA   __static_attributes__rC   r   r   r   r      s     $&< 6666r   r   __main__)numpyr$   r   numbar   numba.typesr   r   r   r   numba.cuda.testingr   r	   r   r   r   rD   mainrC   r   r   <module>rO      sF       5 5 5)#.6 .6b zMMO r   