
    wi                     v   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                    d	eef          e                    d
d          d                         Ze                    dd          d             Ze                    dd          d             ZdZe                    dd          d             Z e	d          Zde_        e                    dd          d             Ze                    dd          d             Ze                    dd          d              Ze                    d!d"          d#             Ze                    d$d"          d%             Ze                    d&d'          d(             Z e	d)          Ze                    d*d'          d+             Zere                    d,d'          d-             Ze                    d.d/          d0             Z e                    d1d2          d3             Z!  e
d4d5g          e             ed6          e           e                    d7d8          d9             Z"e                    d: ee                    d;             Z#e                    d< ee                    d=             Z#e                    d> e                      d?             Z# e	d@          Z$e$                    dAdB          dC             Z%e$                    dDdB          dE             Z&dS )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     m/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/numba/tests/compile_with_pycc.pymultr      s     q5L    get_noneznone()c                      d S r   r   r   r   r   r   r      s    4r   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	    8Or   np_sqrtzf8(f8)c                 *    t          j        |           S r   )npr'   r!   s    r   r.   r.   ;   s    71::r   spacingc                 *    t          j        |           S r   )r0   r1   r!   s    r   
np_spacingr3   ?   s    :a==r   randomzf8(i4)c                     | dk    rt           j                            |            t           j                                        S N)r0   r4   seed)r8   s    r   random_implr9   E   s3    rzz
	t9r   pycc_test_nrtzero_scalarc                 :    t          j        |           }|d         S r6   r0   zeros)nr-   s     r   r;   r;   N   s    
(1++Cr7Nr   
vector_dotc                 X    t          j        d| |           }t          j        ||          S )N   )r0   linspacedot)r?   r   s     r   r@   r@   U   s%    K1a  va||r   r>   z	f8[:](i4)c                 *    t          j        |           S r   r=   )r?   s    r   r>   r>   [   s    8A;;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        |           }|                                D ]\  }}||z  ||<   |S r   )r
   Dictranger*   r0   
zeros_likeitems)r-   dioutkr&   s         r   rI   rI   n   sq    
A38__  1v!
-

C		  1QAJr   internal_str_dictc                     t           j                            t          t                    }| |vrt          |          || <   t          |          S r   )r
   rK   emptyr   r   len)r   rO   s     r   rS   rS   y   s<    
b))Azz1vv!q66Mr   hash_strc                      t          |           S r   hash)r   s    r   rS   rS      s    77Nr   hash_literal_str_Ac                       t          d          S )NArY   r   r   r   rS   rS      s    99r   pycc_test_dynamic_excdo_setitem1zvoid(f8[:, :], f8[:, :])c                     || d d d d f<   d S r   r   r   s     r   r_   r_          AaaadGGGr   do_setitem2c                     || d d d d f<   d S 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      s              ( ( ( ( ( ( ( ( - - - - - - - - - - ( ( ( ( ( (       R
 7Wg&''7L!!  "! (' :x    !  5,    	 8X   r'(( Wl++  ,+ VZ((  )( V[))  *) Y))  *) Y))  *)
 Xx((  )( 
O		}h''  ('  ]]<**  +*
 w$$  %$ |233  43 3

01 2 24 8 8 8   $    ~122  32 "BB|$4$455  65 z22l++,,  -, #RRTT**  +* B&''	
-!;<<  =< -!;<<  =<  r   