
    pi                        d dl Z d dlZd dlmZ d dlmZmZ d dlm	Z	m
Z
mZ d dlmZ d dlmZ  e	d      Zde_        ej                  d	eef      ej                  d
d      d               Zej                  dd      d        Zej                  dd      d        ZdZej                  dd      d        Z e	d      Zde_        ej                  dd      d        Zej                  dd      d        Zej                  dd      d         Zej                  d!d"      d#        Zej                  d$d"      d%        Zej                  d&d'      d(        Z e	d)      Zej                  d*d'      d+        Zerej                  d,d'      d-        Zej                  d.d/      d0        Z ej                  d1d2      d3        Z!  e
d4d5g      e         ed6      e       ej                  d7d8      d9        Z"ej                  d: ee            d;        Z#ej                  d< ee            d=        Z#ej                  d> e             d?        Z# e	d@      Z$e$j                  dAdB      dC        Z%e$j                  dDdB      dE        Z&y)F    N)float32)unicode_typei8)CC
exportmanyexport)has_blas)typedpycc_test_simpleFmultfmultiz
i4(i4, i4)c                     | |z  S N abs     _/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/numba/tests/compile_with_pycc.pymultr      s     q5L    get_noneznone()c                       y r   r   r   r   r   r   r      s    r   divz
f8(f8, f8)c                     | |z  S r   r   )xys     r   r   r      s    q5Lr      squarezi8(i8)c                     | t         z  S r   )_twous    r   r   r   &   s    9r   pycc_test_helperlibpowerz
i8(i8, i8)c                     | |z  S r   r   )r"   vs     r   r$   r$   .   s    6Mr   sqrtzc16(c16)c                 ,    t        j                  |       S r   )cmathr'   r!   s    r   r'   r'   2   s    ::a=r   sizez	i8(f8[:])c                     | j                   S r   )r*   arrs    r   r*   r*   6   s    88Or   np_sqrtzf8(f8)c                 ,    t        j                  |       S r   )npr'   r!   s    r   r.   r.   ;   s    771:r   spacingc                 ,    t        j                  |       S r   )r0   r1   r!   s    r   
np_spacingr3   ?   s    ::a=r   randomzf8(i4)c                     | dk7  rt         j                  j                  |        t         j                  j                         S N)r0   r4   seed)r8   s    r   random_implr9   E   s,    rz
		t99r   pycc_test_nrtzero_scalarc                 6    t        j                  |       }|d   S r6   r0   zeros)nr-   s     r   r;   r;   N   s    
((1+Cr7Nr   
vector_dotc                 \    t        j                  d| |       }t        j                  ||      S )N   )r0   linspacedot)r?   r   s     r   r@   r@   U   s$    KK1a vva|r   r>   z	f8[:](i4)c                 ,    t        j                  |       S r   r=   )r?   s    r   r>   r>   [   s    88A;r   
np_argsortzintp[:](float64[:])c                 ,    t        j                  |       S r   )r0   argsortr,   s    r   rF   rF   `   s    ::c?r   zmultf f4(f4,f4)zmulti i4(i4,i4)zmult f8(f8, f8)dict_usecasezintp[:](intp[:])c                     t        j                         }t        | j                        D ]
  }| |   ||<    t	        j
                  |       }|j                         D ]  \  }}||z  ||<    |S r   )r
   Dictranger*   r0   
zeros_likeitems)r-   dioutkr&   s         r   rI   rI   n   sj    

A388_ 1v!
--
C	 1QAJr   internal_str_dictc                     t         j                  j                  t        t              }| |vrt        |      || <   t        |      S r   )r
   rK   emptyr   r   len)r   rO   s     r   rS   rS   y   s5    

b)Az1v!q6Mr   hash_strc                     t        |       S r   hash)r   s    r   rS   rS      s    7Nr   hash_literal_str_Ac                      t        d      S )NArY   r   r   r   rS   rS      s    9r   pycc_test_dynamic_excdo_setitem1zvoid(f8[:, :], f8[:, :])c                     || d d d d f<   y r   r   r   s     r   r_   r_          AadGr   do_setitem2c                     || d d d d f<   y r   r   r   s     r   rb   rb      ra   r   )'r)   numpyr0   numbar   numba.typesr   r   
numba.pyccr   r   r   numba.tests.supportr	   r
   ccuse_nrtr   r   r   r    r   cc_helperlibr$   r'   r*   r.   r3   r9   cc_nrtr;   r@   r>   rF   rI   rS   	cc_dynexcr_   rb   r   r   r   <module>rn      s3      ( - - (  
 7Wg&'7L! " ( :x  ! 5,   	 8X  '( Wl+ , VZ( ) V[) * Y) * Y) *
 Xx( ) 
O	}h' ( ]]<* +
 w$ % |23 4 3
01 24 8  $  ~12 3 "B|$45 6 z2l+, - #RT* + &'	
-!;< = -!;< =r   