蹑uB
@ d dl Z d dlZd dlZd dlmZmZ d dlmZ edk oedu Z e rpd dl
Z
d dl Z d dl Z d dl
Z
ej ej ej ej e d d dlmZ e j d Zdd Znd dlmZ d d
lmZmZ ed d d d d d
d d Z G d de Z G d d Z d Z e D ] \ Z Z!e!D ]Z" e# e e e" e r e
j$ dS dS ) N)syspath)DistutilsError__main__z..) CCompilerOptc / K t j | i |} |V t j | d S # t j | w xY wN)tempfilemkdtempshutilrmtree)argskwargstmpdirs |/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/distutils/tests/test_ccompiler_opt.pytempdirr sU !42622 "LLL M& ! ! ! ! !FM& ! ! ! !s / A c ( | st | d S r )AssertionError)exprmsgs r assert_r s" & %% % & & )r r )gccclangicciccwmsvc)r r )r r fcc)r )x86x64ppc64ppc64learmhfaarch64s390xnoarchc H e Zd ZdZd
dZd Zd Zd Zeddd Z d S ) FakeCCompilerOptr c H || _ || _ t j | d fi | d S r )fake_trap_filesfake_trap_flagsr __init__)self
trap_files
trap_flagsr r s r r- zFakeCCompilerOpt.__init__) s3 ) ) dD33F33333r c t j d | j | j | S )Nz <<<<
march : {}
compiler : {}
----------------
{}
>>>>
)textwrapdedentformatcc_marchcc_namereportr. s r __repr__zFakeCCompilerOpt.__repr__. s;
VDM4<
?
? @r c t |t sJ t |t sJ | j r4|D ]1}t j | j | r| d 2| j r4|D ]1}t j | j | r| d 2t |d | gt | z S )Nz%source is trapped by a fake interfacez#flag is trapped by a fake interface )
isinstancelistr+ rematch
dist_errorr, zipjoinlen)r. sourcesflagsr srcfs r dist_compilez FakeCCompilerOpt.dist_compile8 s '4(())(%&&''& M
M
M8D 0#66 MOO$KLLL K
K
K8D 0!44 KOO$IJJJ7SXXe__ -G <===r c t j S r )r) fake_infor8 s r dist_infozFakeCCompilerOpt.dist_infoF s ))r F)stderrc d S r )rL r s r dist_logzFakeCCompilerOpt.dist_logI s r N)r r )
__name__
__module__ __qualname__rJ r- r9 rH rK staticmethodrO rN r r r) r) ' s I4 4 4 4
@ @ @ > > > * * * $
\
r r) c e Zd ZdZdZd Zd Zd Zd Zd Z d Z
d Z d d
Z d d Z
d d Zi fd
Zi fdZd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z dS ) _Test_CCompilerOptNc , dt _ d | _ d S )NT)r) conf_nocache_optr8 s r setup_classz _Test_CCompilerOpt.setup_classQ s (,% r c L | j | j dft _ t |i |S Nr )archccr) rJ )r. r r s r noptz_Test_CCompilerOpt.noptU s) &*i"%=" 0000r c P | j s| | _ | j S r )rX r^ r8 s r optz_Test_CCompilerOpt.optY s# y $ DIyr c 4 | j S r )r` r5 r8 s r marchz_Test_CCompilerOpt.march^ s xxzz""r c 4 | j S r )r` r6 r8 s r r6 z_Test_CCompilerOpt.cc_namea s xxzz!!r c |t _ | | dd | dd | dd | dd }t 5 }t
j |d }t |d
5 }| | d d d n # 1 swxY w Y g i }|
|g } | D ]\ }
} t
j |
d d d
} t | }
|
dk rd} nN|
d k r| d } n-dd | z dz } | | || < d d d n # 1 swxY w Y |j | \ }}|r|dgz n|}d |D }t | t k st# fd|D st% d|d ||fS )Nbaselinemindispatchmaxr/ r r0 cpu_baseline cpu_dispatchr/ r0 ztest_targets.cw. r (r; )c r g | ]4}t |t rd d | z dz n|5S )rp r; rq )r< tuplerB ).0tars r
z2_Test_CCompilerOpt.get_targets.. sP
&0U%;%; DC
c ! !
r c 3 K | ]}|v V d S r rN )rt tgtargetss r z1_Test_CCompilerOpt.get_targets.. s' 3S3SaAM3S3S3S3S3S3Sr zE'sources_status' returns different targets than the compiled targets
z != )r) conf_target_groupsr^ getr osr rB openwrite try_dispatchbasenamesplitrC upperappendsources_statusallr )r. targetsgroupsr r` r filerG gflags fake_objectssourcerE gtarglen has_baselinery s @r get_targetsz _Test_CCompilerOpt.get_targetsd s .4+ii J66 J66 zz,33 zz,33
YY %&7<<(899DdC
!A
!
!
!
!
!
!
!
!
!
!
!
!
!
!
!HF ++TF33L!-
%
%
}V,,22377"=4yy199%DDQYY 7==??DD .s 299;;D%%% $t
% % % % % % % % % % % % % % % %* !$ 24 8 g,8E'ZL((g
w<<3x== ( (3S3S3S3S73S3S3S0S0S ( . %ggxx1
s7 1G6C GC G C CGGGc . t dddd }| }| }| || }| || }|||||dz |z |dz |z |dz |z |dz |z fD ] }| |d }| n |rlt |t r/| D ]\ } }
|
dd dvr|
d z || < n(t |t sJ |dd dvr|d z
}|S )
Nr r" r$ r )r! r# r% r _ro z)}$?\.+*$)dictrb r6 r| popr< itemsstr) r. r
map2originrb r6 map_marchmap_cckeyregexkvs r arg_regexz _Test_CCompilerOpt.arg_regex sf
g NN5%00 11 7Iv CK' ! Og % CK& Of $
C JJsD))E ! !%&&
! !KKMM + +DAqv[00#$s7a+ "%--..-:[00SLE r r c | j d i |}|d S | ||| dd | dd }d | }|s't | dk rt
d|z d S t j ||t j st
d|d |d
d S ) Nr/ r r0 ri r; r !expected empty features, not "%s"zdispatch features "
" not match ""rN )
r r^ r| rB cpu_dispatch_namesrC r r> r?
IGNORECASE)r. rg re r r? r` featuress r expectz_Test_CCompilerOpt.expect (((( = Fii! zz,33 zz,33
88C 2 2 4 455 8}}!!$7(B
Fxx 77 .;C88UUUK
r c | j d i |}|d S | ||| dd | dd }d | }|s't | dk rt
d|z d S t j ||t j st
d|d |d
d S ) Nr/ r r0 ri r; r r zbaseline features "r r rN )
r r^ r| rB cpu_baseline_namesrC r r> r? r )r. re rg r r? r` r s r expect_baselinez"_Test_CCompilerOpt.expect_baseline r r c | j d i |}|d S | ||| dd | dd }d | }|s't | dk rt
d|z d S t j || st
d|d |d
d S ) Nr/ r r0 ri r; r z expected empty flags not "%s"zflags "r r rN ) r r^ r| rB cpu_baseline_flagsrC r r> r? )r. re rg r r? r` rE s r expect_flagsz _Test_CCompilerOpt.expect_flags s (((( = Fii! zz,33 zz,33
//1122 5zzQ$3e;
Fxu%% ./4uueee<
r c 6 | j di |}|d S | j d||d|\ }}d | }|s't | dk rt d|z d S t j ||t
j st d|d|d d S ) Nr r r; r z expected empty targets, not "%s"z targets "r r rN )r r rB rC r r> r? r )r. r r r r? r s r expect_targetsz!_Test_CCompilerOpt.expect_targets s (((( = F%T%OgfOOOO
((7## 7||q $6@
Fxw
66 .18%%%@
r c
| j di |}|d S t |t sJ | || \ }}| D ]x\ }}||vrt d|z || } |s%t
| dk rt d|z t j || st d|d| d|d yd S ) Nr z expected to find target "%s"r z+expected to find empty flags in target "%s"r z " flags "r rN ) r r< r r r r rC r> r? )
r. r r r
match_dictr tar_flags match_tar match_flagsrE s
r expect_target_flagsz&_Test_CCompilerOpt.expect_target_flags s' #T^--f--
F*d++,,+'''GG 9&0&6&6&8&8 "I{ ))$2Y> i(E
u::??(E Q 8K//
$n8A 555+++V
r c | j dk rdnd}| j dk rdnd}| }t t |d| j z t t |d|z t t |d| j z t t |d|z d S )Nr" r r r cc_on_cc_is_)r\ r] r` r getattr)r.
wrong_archwrong_ccr` s r test_interfacez!_Test_CCompilerOpt.test_interface s $ W 4 4WW%
$W 4 4WW%hhjjX 122333GCJ!6777888X /00111GCH!455566666r c dD ]j\ }}| || }t | dk sJ t | dk sJ kd S )N))r noner[ )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 +NONErj rk r )r^ rC r r )r. re rg r` s r test_args_emptyz"_Test_CCompilerOpt.test_args_empty s #
7 7 Hh )))IICs--//00A5555s--//00A55555 7 7r c | dk rd S dD ];\ }} | || t d # t $ r Y 8w xY wd S )Nunknown))zunkown_feature - max +minzunknown max min)z#avx2z$vsxr z+excepted an exception for invalid arguments)rb r^ r r )r. re rg s r test_args_validationz'_Test_CCompilerOpt.test_args_validation- s ::<<9 $ $ F#
Hh
xh GGG$%RSSS!
s &A
AAc | ddddd | ddd d
d | d d ddd | d
d
ddd d S )Nz sse vsx neonssevsxneonr )r r" r$ r z)sse41 avx avx2 vsx2 vsx3 neon_vfpv4 asimdzsse41 avx avx2 vsx2 vsx3zneon_vfpv4 asimdz sse neon vsx)re r r" r$ zavx2 vsx3 asimdhp)r r8 s r test_skipz _Test_CCompilerOpt.test_skip; s
U&"
7$ &
^"B
*="B
r c | dddd dD ]F\ }}d|d|d }| |d
|z || | |d |z || Gd
D ]4}|D ]/}| |dd | z 05| dd dD ]}| |d d S )Nzfma3 avx2 asimd vsx3zsse .* sse41 .* fma3.*avx2z
vsx vsx2 vsx3z neon neon_fp16 neon_vfpv4 asimd)r r" r$ )fma3avx2)avx512favx512cdz .* sse42 .* z .*r z.* sse42 .* %s$)x86_gccx86_iccx86_iccwz
.* avx .* %s$)r )r r
avx512_skxz.*%sr; )x86_msvcr vsx vsx2)r# )r neon_fp16
neon_vfpv4asimd)r% )r rB )r. f0f1diffrG ffs r test_impliesz _Test_CCompilerOpt.test_impliesS sm
"0#5
FB .0RR4D )B. t
!
'", t
!
K A
$$R #chhqkk 1 %
UJ 777= O OA ,M N N N N O Or c dD ]}|dk r| dk r!| |ddddd | |ddd d
ddd | |d
dddd | |ddddddd | dddddddd | dd dd | dd d d d d d ! | d"k r$t d#| z d S # t $ r+ | d"k rt d$ Y d S w xY w)%N)rh 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_fp16vx)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& rf 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|-mcpu=a64fx).*.*)r0 r r" r$ r& r% r excepted an exception for %sexcepted no exceptions)r6 r rb r r )r. os r test_args_optionsz$_Test_CCompilerOpt.test_args_options| s " AH}} 6!9!9 KK6b"
KK? e3CBd
KK2"*2#
KKWKLB3,%
E 7j
3B
? KKKDd
zz||y(($2TZZ\\A )( ? ? ?zz||y(($%=>>> )(( ?s AD& &1EEc | dddd| dk rdndddddd
| d
dd | d
dd | dd | dd | dd | dd 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 ppc64_gcc ppc64_clang armhf_gccr% r& r z;-mfp16-format=ieee -mfpu=neon-fp-armv8 -march=armv8-a\+simd)r% r asimdhpz-march=armv8.2-a\+fp16z=-mfp16-format=ieee -mfpu=neon-fp-armv8 -march=armv8.2-a\+fp16) aarch64_gccr asimddpz-march=armv8.2-a\+dotprod)r asimdfhmz -march=armv8.2-a\+fp16\+fp16fmlzasimddp asimdhp asimdfhmz(-march=armv8.2-a\+dotprod\+fp16\+fp16fmlr z-mzvector -march=arch13)r& )r rb r8 s r
test_flagsz _Test_CCompilerOpt.test_flags s@ 5"N!%)ZZ\\U%:%:\\%&:+
T
1V
#?
$F
&C
,
r c , dD ]} | |dddd | dk r$t d| z Z# t $ r* | dk rt d Y w xY wd 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 rb r r r. r s r test_targets_exceptionsz*_Test_CCompilerOpt.test_targets_exceptions s
C CG
C##"Bb $ ::<<9,,(6 E - "
C
C
C::<<9,,()ABBB -,
C3 C Cs AA 1BBc F dD ] }| |ddddd 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 s r test_targets_syntaxz&_Test_CCompilerOpt.test_targets_syntax sN
! ! G>
fD"
?! ! r c | dddddd | dd d
d d d
d | dd ddd
dd | dd dddddddd
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 vsx3vxe2)re 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 asimdvsx2)re rg 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)re 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) re r r r r r" r# r$ r% r r8 s r test_targetsz _Test_CCompilerOpt.test_targets s
- &7{
"F O6
B F.
9-,-! L4#
r c
| ddddddd | d d
d d d
d | dddd| dk rdndddd dD ]*}| d|z ddiddiddiddidd i +| d dd idd idd idd!idd i 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)re 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 */re z.*-O3.*z.*/O3.*z.*/O2.*r )r r r r r z /*@targets baseline $werror */z .*-Werror.*z .*/Werror.*z.*/WX.*)r rb r )r. policys r test_targets_policiesz(_Test_CCompilerOpt.test_targets_policies] sv
6) 1# !
&
* 1#
&
434
-- )(3A 1##
& . F $ $+f4 *I0F +:i2H#D)
%
,M*M0J ]+:i2H % !
r c | dt d ddd | dt d d
d d
ddd d S )NzK
/*@targets $keep_baseline baseline #test_group */
z
$keep_baseline
asimddp sse2 vsx2 avx2 vsx3
avx512f asimdhp
)
test_groupzavx512f avx2 sse2 baselinezvsx3 vsx2 baselinezasimddp asimdhp baseline)r 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
) test_group_1 test_group_2z avx512f avx2 avx sse42 sse41r vsx3zasimdfhm asimddp asimdzasimdfhm asimddp)r r r" r# r$ r% )r r r8 s r test_targets_groupsz&_Test_CCompilerOpt.test_targets_groups s
-&,
/ *&1
r c | ddd | ddddd | d
d ddd
| ddddd 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 r8 s r test_targets_multiz%_Test_CCompilerOpt.test_targets_multi s
-(
# (
(
/V
r r ) rP rQ rR r\ r] rY r^ r` rb r6 r r r r r r r r r r r r r r r r r r r r rN r r rU rU M s D B 1 1 1
# # #" " "* * * X > * * * .0 " 35 .7 7 7 7 7 7
0'O 'O 'OR6? 6? 6?p%
%
%
N C C C<" " "H>
>
>
@J
J
J
X*
*
*
X1
1
1
1
1
r rU c t r0t j d | dz |z | | S t j d | dz |z | | 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()
r )
class_namer\ r] zo class TestCCompilerOpt_{class_name}(_Test_CCompilerOpt):
arch = '{arch}'
cc = '{cc}'
)
is_standaloner2 r3 r4 )r\ r] s r new_testr s X_ .
#:?"
?
#:?"
r r
)%r> r2 r} r r distutils.errorsr rP __package__r
unittest
contextlibr
r r abspathrB dirname__file__
ccompiler_optr contextmanagerr r numpy.distutils.ccompiler_opt
numpy.testingr arch_compilersr) rU r r r\ compilersr] execmainrN r r r sh + + + + + +J&>;$+>
/1111111111111111HOOLDL<4<+A+A4!H!HIIJJJ****** " " "& & & & & ;:::::........
1
1 %
$
$
$
$
$
| $
$
$
Ly
y
y
y
y
y
y
y
v $ &++-- ! !OD) ! ! XXdB
! HMOOOOO r