
    xKg                        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#y)@    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     a/home/alanp/www/video.onchill/myenv/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   )r/   r0   r    s    r   
np_spacingr2   ?   s    ::a=r   randomzf8(i4)c                     | dk7  rt         j                  j                  |        t         j                  j                         S N)r/   r3   seed)r7   s    r   random_implr8   E   s,    rz
		t99r   pycc_test_nrtzero_scalarc                 6    t        j                  |       }|d   S r5   r/   zeros)nr,   s     r   r:   r:   N   s    
((1+Cr7Nr   
vector_dotc                 \    t        j                  d| |       }t        j                  ||      S )N   )r/   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   )r/   argsortr+   s    r   rE   rE   `   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)   r/   
zeros_likeitems)r,   dioutkr%   s         r   rH   rH   n   s`    

A388_1v! 
--
C	1QA Jr   internal_str_dictc                     t         j                  j                  t        t              }| |vrt        |      || <   t        |      S r   )r
   rJ   emptyr   r   len)r   rN   s     r   rR   rR   y   s5    

b)Az1v!q6Mr   hash_strc                     t        |       S r   hash)r   s    r   rR   rR      s    7Nr   hash_literal_str_Ac                      t        d      S )NArX   r   r   r   rR   rR      s    9r   )$r(   numpyr/   numbar   numba.typesr   r   
numba.pyccr   r   r   numba.tests.supportr	   r
   ccuse_nrtr   r   r   r   r   cc_helperlibr#   r&   r)   r-   r2   r8   cc_nrtr:   r?   r=   rE   rH   rR   r   r   r   <module>rf      s      ( - - (  
 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   