a
    Lfjp                  
   @   sH  d dl Z d dlZd dlZd dlmZmZ d dlmZ edkoBedu Z	e	rd dl
Z
d dlZd dlZd dlZejeeeed d dlmZ ejdd Zdd
dZnd dlmZ d dlmZmZ edddddddddZG dd deZG dd dZdd Ze D ]$\Z Z!e!D ]Z"e#ee e" qqe	rDe
$  dS )    N)syspath)DistutilsError__main__z..)CCompilerOptc               	   o   s6   t j| i |}z|V  W t| nt| 0 d S N)tempfilemkdtempshutilrmtree)argskwargstmpdir r   a/var/www/ai-form-bot/venv/lib/python3.9/site-packages/numpy/distutils/tests/test_ccompiler_opt.pytempdir   s    r    c                 C   s   | st |d S r   )AssertionError)exprmsgr   r   r   assert_   s    r   )r   r   )gccclangicciccwmsvc)r   r   )r   r   Zfcc)r   )x86x64ppc64ppc64learmhfaarch64s390xZnoarchc                   @   sD   e Zd ZdZdddZdd Zdd Zdd	 Zed
dddZ	dS )FakeCCompilerOptr   c                 O   s$   || _ || _tj| d fi | d S r   )fake_trap_filesfake_trap_flagsr   __init__)self
trap_files
trap_flagsr   r   r   r   r   r&   )   s    zFakeCCompilerOpt.__init__c                 C   s   t d| j| j|  S )Nz            <<<<
            march    : {}
            compiler : {}
            ----------------
            {}
            >>>>
        )textwrapdedentformatcc_marchcc_namereportr'   r   r   r   __repr__.   s    
zFakeCCompilerOpt.__repr__c                 K   s   t |tsJ t |tsJ | jrD|D ]}t| j|r&| d q&| jrl|D ]}t| j|rN| d qNt|d|gt	| S )Nz%source is trapped by a fake interfacez#flag is trapped by a fake interface )

isinstancelistr$   rematchZ
dist_errorr%   zipjoinlen)r'   sourcesflagsr   srcfr   r   r   dist_compile8   s    zFakeCCompilerOpt.dist_compilec                 C   s   t jS r   )r#   	fake_infor0   r   r   r   	dist_infoF   s    zFakeCCompilerOpt.dist_infoF)stderrc                 G   s   d S r   r   )rA   r   r   r   r   dist_logI   s    zFakeCCompilerOpt.dist_logN)r   r   )
__name__
__module____qualname__r?   r&   r1   r>   r@   staticmethodrB   r   r   r   r   r#   '   s   

r#   c                   @   s   e Zd ZdZdZdd Zdd Zdd Zdd	 Zd
d Z	dd Z
dd Zd5ddZd6ddZd7ddZi fddZi fddZdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 ZdS )8_Test_CCompilerOptNc                 C   s   dt _d | _d S )NT)r#   Zconf_nocache_optr0   r   r   r   setup_classQ   s    z_Test_CCompilerOpt.setup_classc                 O   s   | j | jdft_t|i |S Nr   )archccr#   r?   )r'   r   r   r   r   r   noptU   s    z_Test_CCompilerOpt.noptc                 C   s   | j s|  | _ | j S r   )rH   rM   r0   r   r   r   optY   s    
z_Test_CCompilerOpt.optc                 C   s
   |   jS r   )rN   r-   r0   r   r   r   march^   s    z_Test_CCompilerOpt.marchc                 C   s
   |   jS r   )rN   r.   r0   r   r   r   r.   a   s    z_Test_CCompilerOpt.cc_namec              	      s  |t _| j|dd|dd|dd|ddd}t }tj|d	}t|d
}|	| W d    n1 sx0    Y  g  i }|
|g}	|	D ]t\}
}t|
ddd }t|}|dkrd}n,|dkr|d  }ndd| d  } | |||< qW d    n1 s&0    Y  |j| \}}|rN|dg n|}dd |D }t|t kst fdd|D std| f ||fS )Nbaselinemindispatchmaxr(   r   r)   cpu_baselinecpu_dispatchr(   r)   ztest_targets.cw.   r   (r2   )c                 S   s,   g | ]$}t |tr$d d| d n|qS )r[   r2   r\   )r3   tupler8   ).0tarr   r   r   
<listcomp>   s   z2_Test_CCompilerOpt.get_targets.<locals>.<listcomp>c                 3   s   | ]}| v V  qd S r   r   )r^   tZgtargetsr   r   	<genexpr>       z1_Test_CCompilerOpt.get_targets.<locals>.<genexpr>zM'sources_status' returns different targets than the compiled targets
%s != %s)r#   Zconf_target_groupsrM   getr   osr   r8   openwriteZtry_dispatchbasenamesplitr9   upperappendZsources_statusallr   )r'   targetsgroupsr   rN   r   filer=   ZgflagsZfake_objectssourcer;   ZgtarZglenZhas_baseliner   rb   r   get_targetsd   sH    



(
**z_Test_CCompilerOpt.get_targetsc              	   K   s   t ddddd}|  }|  }|||}|||}|||||d | |d | |d | |d | fD ]}||d }|d url qql|rt|t r| D ]$\}	}
|
dd  dvr|
d	 ||	< qn&t|tsJ |dd  dvr|d	7 }|S )
Nr   r   r    r   )r   r   r!   r   _rZ   z)}$?\.+*$)dictrO   r.   re   popr3   itemsstr)r'   r   Z
map2originrO   r.   Z	map_marchZmap_cckeyregexkvr   r   r   	arg_regex   s6    




z_Test_CCompilerOpt.arg_regexr   c                 K   s   | j f i |}|d u rd S | j|||dd|ddd}d| }|slt|dkrhtd| d S t||tj	std||f d S )	Nr(   r   r)   rT   r2   r   !expected empty features, not "%s"z%dispatch features "%s" not match "%s")
r}   rM   re   r8   cpu_dispatch_namesr9   r   r5   r6   
IGNORECASE)r'   rR   rP   r   r6   rN   featuresr   r   r   expect   s&    


z_Test_CCompilerOpt.expectc                 K   s   | j f i |}|d u rd S | j|||dd|ddd}d| }|slt|dkrhtd| d S t||tj	std||f d S )	Nr(   r   r)   rT   r2   r   r~   z%baseline features "%s" not match "%s")
r}   rM   re   r8   cpu_baseline_namesr9   r   r5   r6   r   )r'   rP   rR   r   r6   rN   r   r   r   r   expect_baseline   s&    


z"_Test_CCompilerOpt.expect_baselinec                 K   s   | j f i |}|d u rd S | j|||dd|ddd}d| }|slt|dkrhtd| d S t||std||f d S )	Nr(   r   r)   rT   r2   r   zexpected empty flags not "%s"zflags "%s" not match "%s")	r}   rM   re   r8   Zcpu_baseline_flagsr9   r   r5   r6   )r'   rP   rR   r   r6   rN   r;   r   r   r   expect_flags   s&    


z_Test_CCompilerOpt.expect_flagsc                 K   s   | j f i |}|d u rd S | jf ||d|\}}d|}|s`t|dkr\td| d S t||tjstd||f d S )Nrn   ro   r2   r   z expected empty targets, not "%s"ztargets "%s" not match "%s")r}   rr   r8   r9   r   r5   r6   r   )r'   rn   ro   r   r6   rs   r   r   r   expect_targets   s    

z!_Test_CCompilerOpt.expect_targetsc           
      K   s   | j f i |}|d u rd S t|ts*J | j||d\}}| D ]^\}}||vr`td| || }	|st|	dkrtd| t||	sDtd||	|f qDd S )Nr   zexpected to find target "%s"r   z+expected to find empty flags in target "%s"z"%s" flags "%s" not match "%s")	r}   r3   ru   rr   rw   r   r9   r5   r6   )
r'   rn   ro   r   Z
match_dictrs   Z	tar_flagsZ	match_tarZmatch_flagsr;   r   r   r   expect_target_flags   s(    z&_Test_CCompilerOpt.expect_target_flagsc                 C   s   | j dkrdnd}| jdkr dnd}|  }tt|d| j   tt|d|   tt|d| j  tt|d|   d S )Nr   r   r   r   Zcc_on_Zcc_is_)rK   rL   rN   r   getattr)r'   Z
wrong_archZwrong_ccrN   r   r   r   test_interface  s    z!_Test_CCompilerOpt.test_interfacec                 C   sH   dD ]>\}}| j ||d}t| dks.J t| dksJ qd S )N))r   nonerJ   )z
none +noneznone - none)z	none -maxz	min - max)z+vsx2 -VSX2zvsx avx2 avx512f -max)z#max -vsx - avx + avx512f neon -MAX z+min -min + max -max -vsx + avx2 -avx2 +NONErU   rV   r   )rM   r9   r   r   )r'   rP   rR   rN   r   r   r   test_args_empty  s    	z"_Test_CCompilerOpt.test_args_emptyc              	   C   sP   |   dkrd S dD ]6\}}z| j||d tdW q tyH   Y q0 qd S )Nunknown))zunkown_feature - max +minzunknown max min)z#avx2z$vsxr   z+excepted an exception for invalid arguments)rO   rM   r   r   )r'   rP   rR   r   r   r   test_args_validation-  s    z'_Test_CCompilerOpt.test_args_validationc                 C   sT   | j dddddd | j ddd	d
dd | j dddddd | j dddddd d S )Nzsse vsx neonssevsxneonr   )r   r   r    r   z)sse41 avx avx2 vsx2 vsx3 neon_vfpv4 asimdzsse41 avx avx2	vsx2 vsx3zneon_vfpv4 asimdzsse neon vsx)rP   r   r   r    zavx2 vsx3 asimdhp)r   r0   r   r   r   	test_skip;  s&    z_Test_CCompilerOpt.test_skipc                 C   s   | j ddddd dD ]@\}}d||f }| j |d| ||d	 | j |d
| ||d	 qdD ]&}|D ]}| j |dd| d qdq\| j ddd dD ]}| j |dd qd S )Nzfma3 avx2 asimd vsx3zsse .* sse41 .* fma3.*avx2zvsx vsx2 vsx3zneon neon_fp16 neon_vfpv4 asimd)r   r   r    )Zfma3avx2)avx512favx512cdz.* sse42 .* %s .*%s$z.* sse42 .* %s$)x86_gccx86_iccx86_iccwz.* avx .* %s$)r   )r   r   Z
avx512_skxz.*%sr2   )x86_msvcr   vsx vsx2)r   )r   Z	neon_fp16Z
neon_vfpv4asimd)r!   )r   r8   )r'   Zf0f1diffr=   ffr   r   r   test_impliesS  s0    
z_Test_CCompilerOpt.test_impliesc                 C   s  dD ]z}|dkr|   dkrq| j|dddddd | j|ddd	d
dddd | j|dddddd | j|dddddddd q| jddddddddd | jddddd z8| jddd d d d d d! |  d"krtd#|   W n* ty   |  d"kr
td$Y n0 d S )%N)rS   nativer   r   z.*cpu_(sse|vsx|neon|vx).cr   )r(   r   r   r    r"   z".*cpu_(sse3|vsx2|neon_vfpv4|vxe).czsse sse2r   zneon neon_fp16vx)r(   r   r   r    r!   r   r"   z.*cpu_(popcnt|vsx3).czsse .* sse41r   zneon neon_fp16 .* asimd .*vx vxe vxe2z=.* xop fma4 .* avx512f .* avx512_knl avx512_knm avx512_skx .*z1.* avx512f .* avx512_knl avx512_knm avx512_skx .*z'.* xop fma4 .* avx512f .* avx512_skx .*z.* asimd asimdhp asimddp .*zvsx vsx2 vsx3 vsx4.*zvx vxe vxe2.*)r   r   r   r   r    r   r"   rQ   zsse sse2 sse3zneon neon_fp16 .* asimd)r   r   r    r!   r   r   r"   z.*cpu_(sse2|vsx2).c)r(   r   r   z..*(-march=native|-xHost|/QxHost|-mcpu=a64fx).*.*)r)   r   r   r    r"   r!   r   excepted an exception for %sexcepted no exceptions)r.   r   rO   r   r   )r'   or   r   r   test_args_options|  s`    

z$_Test_CCompilerOpt.test_args_optionsc                 C   s   | j dddd|  dkrdnddddddd	
 | j d
ddd | j dddd | j ddd | j ddd | j ddd | j ddd d S )Nz'sse sse2 vsx vsx2 neon neon_fp16 vx vxez-msse -msse2z
/arch:SSE2r   r   z-mcpu=power8z"-mfpu=neon-fp16 -mfp16-format=ieeez-mzvector -march=arch12)	r   r   r   r   Z	ppc64_gccZppc64_clang	armhf_gccr!   r"   r   z;-mfp16-format=ieee -mfpu=neon-fp-armv8 -march=armv8-a\+simd)r!   r   Zasimdhpz-march=armv8.2-a\+fp16z=-mfp16-format=ieee -mfpu=neon-fp-armv8 -march=armv8.2-a\+fp16)aarch64_gccr   Zasimddpz-march=armv8.2-a\+dotprod)r   Zasimdfhmz-march=armv8.2-a\+fp16\+fp16fmlzasimddp asimdhp asimdfhmz(-march=armv8.2-a\+dotprod\+fp16\+fp16fmlr   z-mzvector -march=arch13)r"   )r   rO   r0   r   r   r   
test_flags  sF    z_Test_CCompilerOpt.test_flagsc              	   C   sj   dD ]`}z4| j |ddddd |  dkr:td|   W q tyb   |  dkr^tdY q0 qd S )N)zbla blaz
/*@targetsz/*@targets */z/*@targets unknown */z"/*@targets $unknown_policy avx2 */z!/*@targets #unknown_group avx2 */z/*@targets $ */z/*@targets # vsx */z/*@targets #$ vsx */z/*@targets vsx avx2 ) */z/*@targets vsx avx2 (avx2 */z/*@targets vsx avx2 () */z!/*@targets vsx avx2 ($autovec) */z/*@targets vsx avx2 (xxx) */z!/*@targets vsx avx2 (baseline) */r   )r   r    r   r"   r   r   r   )r   rO   r   r   r'   rn   r   r   r   test_targets_exceptions  s    
z*_Test_CCompilerOpt.test_targets_exceptionsc              	   C   s$   dD ]}| j |dddddd qd S )N)z+/*@targets $keep_baseline sse vsx neon vx*/z+/*@targets,$keep_baseline,sse,vsx,neon vx*/z+/*@targets*$keep_baseline*sse*vsx*neon*vx*/zs
            /*
            ** @targets
            ** $keep_baseline, sse vsx,neon, vx
            */
            z
            /*
            ************@targets****************
            ** $keep_baseline, sse vsx, neon, vx
            ************************************
            */
            z
            /*
            /////////////@targets/////////////////
            //$keep_baseline//sse//vsx//neon//vx
            /////////////////////////////////////
            */
            zj
            /*
            @targets
            $keep_baseline
            SSE VSX NEON VX*/
            r   r   r   r   r   )r   r   r    r"   r   r   r   r   r   r   test_targets_syntax  s    
z&_Test_CCompilerOpt.test_targets_syntaxc                 C   sh   | j ddddddd | j dd	d
ddddd | j dd	dddddd | j dd	ddddddddd
 d S )Nz
            /*@targets
                sse sse2 sse41 avx avx2 avx512f
                vsx vsx2 vsx3 vsx4
                neon neon_fp16 asimdhp asimddp
                vx vxe vxe2
            */
            zavx vsx2 asimd vx vxeavx512f avx2zasimddp asimdhpz	vsx4 vsx3vxe2)rP   r   r    r   r"   z
            /*@targets
                sse41 avx avx2 avx512f
                vsx2 vsx3 vsx4
                asimd asimdhp asimddp
                vx vxe vxe2
            */
            r   z"sse41 avx2 vsx2 asimd asimddp vxe2z
avx2 sse41asimddp asimdZvsx2)rP   rR   r   r    r   r"   z
            /*@targets
                sse2 sse41 avx2 avx512f
                vsx2 vsx3 vsx4
                neon asimdhp asimddp
                vx vxe vxe2
            */
            z).*(avx2|avx512f|vsx3|vsx4|asimddp|vxe2).cz
sse41 sse2zasimdhp neonzvxe vx)rP   r(   r   r   r    r"   z
            /*@targets
                sse sse2 avx fma3 avx2 avx512f avx512cd
                vsx vsx2 vsx3
                neon neon_vfpv4 neon_fp16 neon_fp16 asimd asimdhp
                asimddp asimdfhm
            */
            z#avx512cd avx512f avx2 fma3 avx sse2zavx512cd avx2 avx sse2zvsx3 vsx2 vsx	vsx3 vsx2z8asimdfhm asimddp asimdhp asimd neon_vfpv4 neon_fp16 neonzasimdfhm asimddp asimdhp asimd)	rP   r   r   r   r   r   r   r    r!   r   r0   r   r   r   test_targets  s<    z_Test_CCompilerOpt.test_targetsc              
   C   s   | j dddddddd | j d	d
ddddd | j dddd|  dkrJdnddddd dD ]2}| jd| ddiddiddiddiddid q\| jdddiddidd idd!iddid d S )"Nz
            /*@targets
                $keep_baseline
                sse2 sse42 avx2 avx512f
                vsx2 vsx3
                neon neon_vfpv4 asimd asimddp
                vx vxe vxe2
            */
            zsse41 avx2 vsx2 asimd vsx3 vxezavx512f avx2 sse42 sse2r   zasimddp asimd neon_vfpv4 neonr   zvxe2 vxe vx)rP   r   r   r    r!   r"   z
            /*@targets
                $keep_baseline $keep_sort
                avx512f sse42 avx2 sse2
                vsx2 vsx3
                asimd neon neon_vfpv4 asimddp
                vxe vxe2
            */
            zavx512f sse42 avx2 sse2r   zasimd neon neon_vfpv4 asimddpzasimd asimddpzvxe vxe2)r   r   r    r!   r"   z
            /*@targets
                $keep_baseline $keep_sort $autovec
                avx512f avx2 sse42 sse41 sse2
                vsx3 vsx2
                asimddp asimd neon_vfpv4 neon
            */
            zavx512f avx2 sse42 sse41 sse2r   zavx512f avx2 sse2r   )r   r   r   r   r   r    r!   )z$maxoptz$autovecz/*@targets baseline %s */rP   z.*-O3.*z.*/O3.*z.*/O2.*r   )r   r   r   r   r   z/*@targets baseline $werror */z.*-Werror.*z.*/Werror.*z.*/WX.*)r   rO   r   )r'   policyr   r   r   test_targets_policies]  sT    		
z(_Test_CCompilerOpt.test_targets_policiesc              	   C   s>   | j dtdddddd | j dtd	d
ddddddd d S )NzK
            /*@targets $keep_baseline baseline #test_group */
            z
                    $keep_baseline
                    asimddp sse2 vsx2 avx2 vsx3
                    avx512f asimdhp
                )Z
test_groupzavx512f avx2 sse2 baselinezvsx3 vsx2 baselinezasimddp asimdhp baseline)ro   r   r   r    z
            /*@targets
             * sse42 avx avx512f
             * #test_group_1
             * vsx2
             * #test_group_2
             * asimddp asimdfhm
            */
            z@
                    VSX2 vsx3 asimd avx2 SSE41
                z@
                    vsx2 vsx3 asImd aVx2 sse41
                )Ztest_group_1Ztest_group_2zavx512f avx2 avx sse42 sse41r   vsx3zasimdfhm asimddp asimdzasimdfhm asimddp)ro   r   r   r   r    r!   )r   ru   r0   r   r   r   test_targets_groups  s*    	z&_Test_CCompilerOpt.test_targets_groupsc                 C   sP   | j dddd | j dddddd	 | j d
ddddd | j dddddd d S )Nzm
            /*@targets
                (avx512_clx avx512_cnl) (asimdhp asimddp)
            */
            z\(avx512_clx avx512_cnl\)z\(asimdhp asimddp\))r   r    z
            /*@targets
                f16c (sse41 avx sse42) (sse3 avx2 avx512f)
                vsx2 (vsx vsx3 vsx2)
                (neon neon_vfpv4 asimd asimdhp asimddp)
            */
            zavx512f f16c avxr   r   )r   r   r   r    z
            /*@targets $keep_sort
                (sse41 avx sse42) (sse3 avx2 avx512f)
                (vsx vsx3 vsx2)
                (asimddp neon neon_vfpv4 asimd asimdhp)
                (vx vxe vxe2)
            */
            zavx avx512fr   )r   r   r    r"   zz
            /*@targets $keep_sort
                fma3 avx2 (fma3 avx2) (avx2 fma3) avx2 fma3
            */
            zfma3 avx2 \(fma3 avx2\)r   )r   r   r   r   r   r0   r   r   r   test_targets_multi  s2    
z%_Test_CCompilerOpt.test_targets_multi)r   )r   )r   )rC   rD   rE   rK   rL   rI   rM   rN   rO   r.   rr   r}   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rG   M   s6   ,


	)8'$@L,rG   c                 C   s@   t r"tdj| d | | |dS tdj| d | | |dS )Na      class TestCCompilerOpt_{class_name}(_Test_CCompilerOpt, unittest.TestCase):
        arch = '{arch}'
        cc   = '{cc}'
        def __init__(self, methodName="runTest"):
            unittest.TestCase.__init__(self, methodName)
            self.setup_class()
    rs   )
class_namerK   rL   zo    class TestCCompilerOpt_{class_name}(_Test_CCompilerOpt):
        arch = '{arch}'
        cc   = '{cc}'
    )is_standaloner*   r+   r,   )rK   rL   r   r   r   new_test  s    

r   )r   )%r5   r*   rf   r   r   distutils.errorsr   rC   __package__r   Zunittest
contextlibr   r
   rl   abspathr8   dirname__file__Zccompiler_optr   contextmanagerr   r   Znumpy.distutils.ccompiler_optZnumpy.testingru   Zarch_compilersr#   rG   r   rw   rK   Z	compilersrL   execmainr   r   r   r   <module>   sF     
&     @