U
|
O ±ds ã
|
@ sH d dl Z d dlZd dlZd dlmZmZ d dlmZ edkoBedkZ e r®d dl
|
Z
|
d dl Z d dl Z d dl Z ej e e e e¡d¡¡¡ d dlmZ e jdd Zdd
|
d Zn d dlmZ d d lmZmZ ed d dddddddZG dd deZ G dd dZ dd Z e ¡ D ]$\Z Z!e!D ]Z"e#e e e" q qe rDe
|
$¡ dS )é N)ÚsysÚpath)ÚDistutilsErrorÚ__main__z..)Ú CCompilerOptc o s( t j| |}z
|
|V W 5 t |¡ X d S ©N)ÚtempfileÚmkdtempÚshutilÚrmtree)ÚargsÚkwargsÚtmpdir© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\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 )Úx86Úx64Úppc64Úppc64leÚarmhfÚaarch64Ús390xZnoarchc @ sD e Zd ZdZdddZdd Zdd Zdd Zed
|
d d d Z dS )ÚFakeCCompilerOptr c O s || _ || _tj| d f| d S r )Úfake_trap_filesÚfake_trap_flagsr Ú__init__)ÚselfÚ
|
trap_filesÚ
|
trap_flagsr r r r r r& ) s zFakeCCompilerOpt.__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
|
|
ùzFakeCCompilerOpt.__repr__c K s t |tstt |ts t| jrD|D ] }t | j|¡r&| d¡ q&| jrl|D ] }t | j|¡rN| d¡ qNt|d |¡gt
|
| S )Nz%source is trapped by a fake interfacez#flag is trapped by a fake interfaceú ) Ú
|
isinstanceÚlistr r$ ÚreÚmatchZ
|
dist_errorr% ÚzipÚjoinÚlen)r' ÚsourcesÚflagsr ÚsrcÚfr r r Ú dist_compile8 s z FakeCCompilerOpt.dist_compilec C s t jS r )r# Ú fake_infor0 r r r Ú dist_infoF s zFakeCCompilerOpt.dist_infoF)Ústderrc G s d S r r )rA r r r r Údist_logI s zFakeCCompilerOpt.dist_logN)r r )
|
Ú__name__Ú
|
__module__Ú __qualname__r? r& r1 r> r@ Ú staticmethodrB r r r r r# ' s
|
|
r# c @ sê e Zd ZdZdZdd Zdd Zdd Zdd Zd
|
d Z d d Z
|
dd Z d5ddZ d6ddZ d7ddZi fddZi fddZdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Z d3d4 Z dS )8Ú_Test_CCompilerOptNc C s dt _d | _d S )NT)r# Z conf_nocacheÚ_optr0 r r r Ú setup_classQ s z _Test_CCompilerOpt.setup_classc O s | j | jdft_t||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
|
| ¡ jS r )rN r- r0 r r r Úmarch^ s z_Test_CCompilerOpt.marchc C s
|
| ¡ jS r )rN r. r0 r r r r. a s z_Test_CCompilerOpt.cc_namec
|
sx |t _| j| dd¡| dd¡| dd¡| dd¡d}t Â}tj |d ¡}t|d
|
}| |¡ W 5 Q R X g i }|
|
|g¡} | D ]t\}
|
} t |
|
¡ d ¡d d
} t | } | dkr¼d} n,| d krÒ| d ¡ } ndd | ¡ d ¡ } | ¡ | || < qW 5 Q R X |j| \}}|r$|dg n|}dd |D }t |t ks`t fdd|D sptd| f ||fS )NÚbaselineÚminÚdispatchÚmaxr( r r) ©Ú cpu_baselineÚ cpu_dispatchr( r) ztest_targets.cÚwÚ.é éÿÿÿÿr ú(r2 ú)c S s, g | ]$}t |tr$d d |¡ d n|qS )r[ r2 r\ )r3 Útupler8 )Ú.0Útarr r r Ú
|
<listcomp> s ÿz2_Test_CCompilerOpt.get_targets.<locals>.<listcomp>c 3 s | ]}| kV qd S r r )r^ Út©Zgtargetsr r Ú <genexpr> s z1_Test_CCompilerOpt.get_targets.<locals>.<genexpr>zM'sources_status' returns different targets than the compiled targets
|
%s != %s)r# Zconf_target_groupsrM Úgetr Úosr r8 ÚopenÚwriteZ try_dispatchÚbasenameÚsplitr9 ÚupperÚappendZsources_statusÚallr )r' ÚtargetsÚgroupsr rN r Úfiler= ZgflagsZ fake_objectsÚsourcer; ZgtarZglenZ has_baseliner rb r Ú get_targetsd sH
|
|
|
|
ü
|
þ*ÿÿz _Test_CCompilerOpt.get_targetsc K sò t ddddd}| ¡ }| ¡ }| ||¡}| ||¡}|||||d | |d | |d | |d | fD ] }| |d ¡}|d k rl qql|rît|t rÈ| ¡ D ]$\} }
|
|
|
dd
dkr |
|
d || < q n&t|tsÖt|dd
dkrî|d 7 }|S )
|
Nr r r r )r r r! r Ú_rZ z)}$?\.+*ú$) ÚdictrO r. rd Úpopr3 ÚitemsÚstrr ) r' r Z
|
map2originrO r. Z map_marchZmap_ccÚkeyÚregexÚkÚvr r r Ú arg_regex s> ü
|
|
|
|
û
|
z _Test_CCompilerOpt.arg_regexr c K s | j f |}|d krd S | j||| dd¡| dd¡d}d | ¡ ¡}|sht|dkrdtd| d S t ||tj ¡std||f d S ) Nr( r r) rT r2 r ú!expected empty features, not "%s"z%dispatch features "%s" not match "%s")
|
r| rM rd 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 |}|d krd S | j||| dd¡| dd¡d}d | ¡ ¡}|sht|dkrdtd| d S t ||tj ¡std||f d S ) Nr( r r) rT r2 r r} z%baseline features "%s" not match "%s")
|
r| rM rd 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 |}|d krd S | j||| dd¡| dd¡d}d | ¡ ¡}|sht|dkrdtd| d S t ||¡std||f d S ) Nr( r r) rT r2 r z expected empty flags not "%s"zflags "%s" not match "%s") r| rM rd r8 Zcpu_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 |}|d krd S | jf ||d|\}}d |¡}|s\t|dkrXtd| d S t ||tj¡s|td||f d S )N©rm rn r2 r z expected empty targets, not "%s"ztargets "%s" not match "%s")r| rq r8 r9 r r5 r6 r )r' rm rn r r6 rr r r r Úexpect_targetsî s
|
ÿ
|
ÿz!_Test_CCompilerOpt.expect_targetsc
|
K s¤ | j f |}|d krd S t|ts&t| j||d\}}| ¡ D ]^\}}||kr\td| || } |st| dkrtd| t || ¡s@td|| |f q@d 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 rt r rq rv r9 r5 r6 )
|
r' rm rn r Z
|
match_dictrr Z tar_flagsZ match_tarZ match_flagsr; r r r Úexpect_target_flagsÿ s( ÿ ÿ ÿz&_Test_CCompilerOpt.expect_target_flagsc C s | j dkrdnd}| jdkr dnd}| ¡ }tt|d| j tt|d| tt|d| j tt|d| d S )Nr r r r Zcc_on_Zcc_is_)rK rL rN r Úgetattr)r' Z
|
wrong_archZwrong_ccrN r r r Útest_interface s z!_Test_CCompilerOpt.test_interfacec C sH dD ]>\}}| j ||d}t| ¡ dks.tt| ¡ dkstqd S )N))r ÚnonerJ )z
|
none +nonez none - none)z none -maxz min - max)z +vsx2 -VSX2zvsx 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~ )r' rP rR rN r r r Útest_args_empty s z"_Test_CCompilerOpt.test_args_emptyc C sR | ¡ dkrd S dD ]8\}}z| j||d tdW q tk
|
rJ Y qX qd S )NÚunknown))zunkown_feature - max +minzunknown 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 dddddd | j ddd d
|
dd | j d d dddd | j d d dddd d S )Nz sse vsx neonÚsseÚvsxÚneonr )r r r r z)sse41 avx avx2 vsx2 vsx3 neon_vfpv4 asimdzsse41 avx avx2ú vsx2 vsx3zneon_vfpv4 asimdz sse neon vsx)rP r r r zavx2 vsx3 asimdhp)r r0 r r r Ú test_skip; s8 þû þ þz _Test_CCompilerOpt.test_skipc C s® | j ddddd dD ]@\}}d||f }| j |d| ||d | j |d
|
| ||d qd D ]&}|D ] }| j |d d |¡ d qdq\| j ddd dD ]}| j |dd qd S )Nzfma3 avx2 asimd vsx3zsse .* sse41 .* fma3.*avx2z vsx vsx2 vsx3z neon neon_fp16 neon_vfpv4 asimd)r r r )©Zfma3Ú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' Zf0Úf1Zdiffr= Úffr r r Ú test_impliesS s4 û þ þ ÿ
|
z _Test_CCompilerOpt.test_impliesc C s dD ]z}|dkr | ¡ dkr q| j|dddddd | j|ddd d
|
ddd d | j|d ddddd | j|dddddddd q| jddddddddd | jdd ddd z6| jdd d d d d d! | ¡ d"kràtd#| ¡ W n, tk
|
r | ¡ d"kr
|
td$Y nX 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).czsse sse2r zneon neon_fp16Úvx)r( r r r r! r r" z.*cpu_(popcnt|vsx3).cz sse .* sse41r zneon 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 .*zvsx vsx2 vsx3 vsx4.*z vx vxe vxe2.*)r r r r r r r" rQ z sse sse2 sse3zneon neon_fp16 .* asimd)r r r r! r r r" z.*cpu_(sse2|vsx2).c)r( r r z".*(-march=native|-xHost|/QxHost).*ú.*)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 dddd| ¡ dkr dndddddd d
|
|
| j d dd d | j dddd | j ddd | j ddd | j ddd | j ddd d S ) Nz'sse sse2 vsx vsx2 neon neon_fp16 vx vxez -msse -msse2z
|
/arch:SSE2r r z -mcpu=power8z -maltivec -mvsx -mpower8-vectorz"-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© Zasimdhpz-march=armv8.2-a\+fp16z=-mfp16-format=ieee -mfpu=neon-fp-armv8 -march=armv8.2-a\+fp16)Ú aarch64_gccr© Zasimddpz-march=armv8.2-a\+dotprod)rª Zasimdfhmz -march=armv8.2-a\+fp16\+fp16fmlzasimddp asimdhp asimdfhmz(-march=armv8.2-a\+dotprod\+fp16\+fp16fmlr£ z-mzvector -march=arch13)r" )r rO r0 r r r Ú
|
test_flags´ sL ÷ ýý ÿ þþþz _Test_CCompilerOpt.test_flagsc C sl dD ]b}z4| j |ddddd | ¡ dkr:td| ¡ W q tk
|
rd | ¡ dkr`tdY qX qd S )N)zbla 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' rm r r r Útest_targets_exceptionsÛ s þ
|
ÿ z*_Test_CCompilerOpt.test_targets_exceptionsc C s$ dD ]}| j |dddddd qd 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 ddddddd | j dd d
|
d d d dd | j dd ddd ddd | j dd ddddddddd
|
d S ) NzÑ
|
/*@targets
|
sse sse2 sse41 avx avx2 avx512f
|
vsx vsx2 vsx3 vsx4
|
neon neon_fp16 asimdhp asimddp
|
vx vxe vxe2
|
*/
|
zavx vsx2 asimd vx vxeú avx512f avx2zasimddp 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 asimdZvsx2)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 neonzvxe 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 sse2zavx512cd 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 sL õ ö ôïz _Test_CCompilerOpt.test_targetsc
|
C s¾ | j dddddddd | j d d
|
d d d dd | j dddd| ¡ dkrJdnddddd dD ]2}| jd| ddiddiddiddidd id q\| jd dd idd idd idd!idd id 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 vxezavx512f 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
|
*/
|
zavx512f sse42 avx2 sse2r z asimd neon neon_vfpv4 asimddpz asimd asimddpzvxe 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 zavx512f 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] s\ ð ñ
|
ÿï ü üz(_Test_CCompilerOpt.test_targets_policiesc C s> | j dtdddddd | j dtd d
|
d d d dddd d S )NzK
|
/*@targets $keep_baseline baseline #test_group */
|
z
|
$keep_baseline
|
asimddp sse2 vsx2 avx2 vsx3
|
avx512f asimdhp
|
)Z
|
test_groupzavx512f avx2 sse2 baselinezvsx3 vsx2 baselinezasimddp asimdhp baseline)rn 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³ Úvsx3zasimdfhm asimddp asimdzasimdfhm asimddp)rn r r r r r! )r rt r0 r r r Útest_targets_groups© s* ÿó üèz&_Test_CCompilerOpt.test_targets_groupsc C sP | j dddd | j dddddd | j d
|
d ddd d | j dddddd 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)
|
*/
|
zavx512f 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
|
*/
|
zfma3 avx2 \(fma3 avx2\)r )r r r r r® r0 r r r Útest_targets_multiÕ s4 ù
|
õô øz%_Test_CCompilerOpt.test_targets_multi)r )r )r ) rC rD rE rK rL rI rM rN rO r. rq 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 | | |dS t d¡j| d | | |dS )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()
|
rr )Ú
|
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* re r r Údistutils.errorsr rC Ú __package__r» ZunittestÚ
|
contextlibr r
|
rk Úabspathr8 ÚdirnameÚ__file__Z ccompiler_optr Úcontextmanagerr r Z numpy.distutils.ccompiler_optZ numpy.testingrt Zarch_compilersr# rG r¼ rv rK Z compilersrL ÚexecÚmainr r r r Ú<module> sF
|
ø & @
|
