1j
: d dl Z d dlZd dlZd dlZd dlmZmZmZ d dlZ d dl
Z
d dl Z d dl Z d dlZd Z
d Z G d d Zej e j dk d G d
d Ze j d Ze j d
Z ej Z ej deej Zej epe pe d G d de Z ej deej Zej e pe d G d de Z ej deej Z ej e pe d G d de Z ej deej Z ej e pe d G d d e Z dS ) N)__cpu_features____cpu_baseline____cpu_dispatch__c
d}t | t | }} | |k rd S t t dd } t d 5 }| d }d d d n # 1 swxY w Y n&# t
$ r}t | }Y d }~nd }~ww xY w dd l} |j dgt d
} | } n&# t
$ r}t | } Y d }~nd }~ww xY wdd l
}
|
d |d |d
| dd } t d|d| d|d| )NT'
/proc/cpuinfoi r /bin/true1 LD_SHOW_AUXVenvz
###########################################
### Extra debugging information
###########################################
-------------------------------------------
--- NumPy Detections
-------------------------------------------
zk
-------------------------------------------
--- SYS / CPUINFO
-------------------------------------------
zl....
-------------------------------------------
--- SYS / AUXV
-------------------------------------------
)prefixzFailure Detection
NAME: 'z '
ACTUAL: z
DESIRED: )
strr replaceopenread Exception
subprocess check_outputdictdecodetextwrapindentAssertionError) actualdesiredfname__tracebackhide__detectedfdcpuinfoerrr auxvr error_reports v/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/core/tests/test_cpu_features.pyassert_features_equalr*
s &kk3w<z.AbstractTest.test_features..L s% @ @ @!T]]1 - - @ @ @r2 ) r: features_groupsitemsr* r getallfeaturesr= )r9 gnamerE
test_features feature_namer= npy_haves ` r) rG zAbstractTest.test_featuresI s # 3 9 9 ; ; Z ZOE8 @ @ @ @x @ @ @M !"2"6u"="=s=?Q?QSX Y Y Y Y M D DL}}\22H'+L99H !(Hl C C C C D Dr2 c | j || }t |t r || j v S |D ]}|| j v r dS dS )NTF) features_maprC
isinstancer features_flags)r9 rH map_namesr? s r) r= zAbstractTest.cpu_haveT sh %)), EE i % % 4 33 3 AD'''tt (ur2 c : | | | _ d S r6 )get_cpuinfo_itemrM )r9 magic_keys r) load_flags_cpuinfoz AbstractTest.load_flags_cpuinfo] s " 3 3I > >r2 c t }t d 5 }|D ]}| | sd | dd D }t | dk r?| |d } d d d n # 1 swxY w Y |S )Nr c 6 g | ]}| S r7 r- r> ss r) r@ z1AbstractTest.get_cpuinfo_item..f s E E EQqwwyy E E Er2 : )setr
startswithr. lenunionupper)r9 rQ valuesr$ line flags_values r) rP z AbstractTest.get_cpuinfo_item` s
/
"
" Jb
J
Jy11 E E$**S!2D2D E E E {##q(( #\\+a.*>*>*@*@*F*F*H*H I IF
J J J J J J J J J J J J J J J J
s BB99B= B=c t j dgt d }| d D ]}| d sd | dd D }t | d
k r[| j |d | _ d S ) Nr
r r r
s AT_HWCAPc 6 g | ]}| S r7 rU rV s r) r@ z0AbstractTest.load_flags_auxv..p s @@@17799@@@r2 :rY rZ )
r r r r. r\ r] rM r^ r_ r )r9 r' at hwcap_values r) load_flags_auxvz AbstractTest.load_flags_auxvk s & }$C:P:P:PQQQ**U## B==--
@@bhhtQ.?.?@@@K; 1$$&*&9&?&? N((**113399;;' '# r2 N)__name__
__module__ __qualname__rE rA rK r[ rM r: rG r= rR rP ri r7 r2 r) r4 r4 @ s HOLSUUN
D D D ? ? ? r2 r4
emscriptenzsThe subprocess module is not available on WASM platforms and therefore this test class cannot be properly executed.reasonc e Zd Z ej e j Ze j
Z e j
dd Ze j
dd Z eeddd Zd eD Z ee dk rdned Z ee dk rdned ZdZ ej d d
Zd Z d d
Z d Z d Z ej ! dddg d Z"ej # e d ej ! dddg d Z$ej # e d d Z%d Z&dS ) TestEnvPrivationNPY_ENABLE_CPU_FEATURESNNPY_DISABLE_CPU_FEATUREST)cwdcapture_outputtextcheckc , g | ]}t | |S r7 )r )r> feats r) r@ zTestEnvPrivation. s3 1A$1G r2 r z
def main():
from numpy.core._multiarray_umath import __cpu_features__, __cpu_dispatch__
detected = [feat for feat in __cpu_dispatch__ if __cpu_features__[feat]]
print(detected)
if __name__ == "__main__":
main()
)autousec | | d }|dz }| | j || _ d S )Nruntime_test_scriptz_runtime_detect.py)mktemp
write_textSCRIPTfile)r9 tmp_path_factoryr s r) setup_classz TestEnvPrivation.setup_class sA &&'<== $$ $$$ r2 c ` t j t j | j gfd| j i| j S )Nr )r runsys
executabler r SUBPROCESS_ARGSr8 s r) _runzTestEnvPrivation._run s> ~
^TY ' " r2 Failed to generate errorc > | J | # t j $ rm}d| d|j }t j ||j s
J | d| d|j }t j ||j s
J | Y d }~d S d }~ww xY w)NFz
Expected: z
Got: zExpected error of type: z; see full error:
)r r CalledProcessErrorstderrresearch)r9 msgerr_type no_error_msgeassertion_messages r)
_expect_errorz TestEnvPrivation._expect_error s ' IIKKK
', & &5 , D D D CS C C C C 9S!(++ > >-> > >+&8 & & 8& &
9Xqx00 C C2C C C0 C C C C C C Ds B A"BB c L t j | _ dS )z$Ensure that the environment is resetN)osenvironcopyr r8 s r) setup_methodz TestEnvPrivation.setup_method s :??$$r2 c > | }t |j }|t j d |d }|| j d< | }t |j }t
| |hk sJ t | dk rt j d d | | j d< | }t |j }t
| t
| k sJ dS )z
Ensure that when selecting `NPY_ENABLE_CPU_FEATURES`, only the
features exactly specified are dispatched.
Nz6No dispatchable features outside of baseline detected.r rr rZ z'Only one non-baseline feature detected.,) r r1 stdoutpytestskipr r[ r] join)r9 r0 non_baseline_featuresfeatureenabled_featuress r) test_runtime_feature_selectionz/TestEnvPrivation.test_runtime_feature_selection s iikk -cj 9 9 ( KH
(* /6*+iikk(44 #$$ 1111 $ % % ) ) KA B B B /2hh7L.M.M*+iikk(44 #$$,A(B(BBBBBr2 zenabled, disabled)r r )r samec b || j d< || j d< d}d}| || dS )zk
Ensure that when both environment variables are set then an
ImportError is thrown
rr rs z9Both NPY_DISABLE_CPU_FEATURES and NPY_ENABLE_CPU_FEATURES ImportErrorNr r )r9 enableddisabledr r s r) test_both_enable_disable_setz-TestEnvPrivation.test_both_enable_disable_set sD /6*+/7+,I 3)))))r2 zANPY_*_CPU_FEATURES only parsed if `__cpu_dispatch__` is non-emptyrn actionENABLEDISABLEc z d}d|z | j d| d< d| d|dz d| d }d
}| || d S ) z
Test that an error is thrown if the environment variables are too long
to be processed. Current limit is 1024, but this may change later.
i tNPY_
_CPU_FEATURESz$Length of environment variable 'NPY_z_CPU_FEATURES' is rY z, only z accepted RuntimeErrorNr )r9 r MAX_VAR_LENGTHr r s r) test_variable_too_longz'TestEnvPrivation.test_variable_too_long s 14~1E----.
D6
D
D !
D
D*8
D
D
D " 3)))))r2 c | j t j d | j }|| j d< d| d}d}| || dS )z
Test that a RuntimeError is thrown if an impossible feature-disabling
request is made. This includes disabling a baseline feature.
N.There are no unavailable features to test withrs z You cannot disable CPU feature 'z1', since it is part of the baseline optimizationsr )
BASELINE_FEATr r r r )r9 bad_featurer r s r) test_impossible_feature_disablez0TestEnvPrivation.test_impossible_feature_disable
sq % KH I I I( /:+,
1{
1
1
1 " 3)))))r2 c | j t j d | j }|| j d< d| d}d}| || | d| j }|| j d< d| d}| || dS )z
Test that a RuntimeError is thrown if an impossible feature-enabling
request is made. This includes enabling a feature not supported by the
machine, or disabling a baseline optimization.
Nr rr z!You cannot enable CPU features \(z1\), since they are not supported by your machine.r r, )UNAVAILABLE_FEATr r r r r )r9 r r r featss r) test_impossible_feature_enablez/TestEnvPrivation.test_impossible_feature_enable% s ( KH I I I+ .9*+
6
6
6
6 " 3))) 66$"466.3*+
1
1
1
1
3)))))r2 )r )'rj rk rl pathlibPath__file__parentresolvert r r r r pop_enable_disabler r r unavailable_featsr] r r r r r fixturer r r r r mark parametrizer skipifr r r r7 r2 r) rq rq v sc ',x
'
/
/
1
1C
*//
Cjnn 6==Gz~~ 8$??Hds4d$OOOO ) %&&!++
q
! C 0 1 1Q 6 6DD=
* [
. * * *"* * * * *r2 rq linuxcygwinz^(amd64|x86|i386|i686)zOnly for Linux and x86c e Zd Zg dZ eg dg dg dg dg dg dg d Z ed
d d d
dddddd
Zd ZdS )Test_X86_Features) MMXSSESSE2SSE3SSSE3SSE41POPCNTSSE42AVXF16CXOPFMA4FMA3AVX2AVX512FAVX512CDAVX512ERAVX512PF AVX5124FMAPS AVX5124VNNIWAVX512VPOPCNTDQAVX512VLAVX512BWAVX512DQ
AVX512VNNI
AVX512IFMA
AVX512VBMI AVX512VBMI2 AVX512BITALG
AVX512FP16)r r r r )r r r r r r r )r r r r r )r r r r r r )r r r r r r r ) r r r r r r r r r r r ) r r r r r r r r r r r r )
AVX512_KNL
AVX512_KNM
AVX512_SKX
AVX512_CLX
AVX512_CNL
AVX512_ICL
AVX512_SPRPNISSE4_1SSE4_2FMA AVX512_VNNI
AVX512_BITALG AVX512_VBMI2
AVX512_4FMAPS
AVX512_4VNNIWAVX512_VPOPCNTDQ)
r r r r r r r r r r c 0 | d d S )Nflags)rR r8 s r) r: z Test_X86_Features.load_flagsc s (((((r2 N)rj rk rl rE r rA rK r: r7 r2 r) r r D s H dDDD9 9 9PPP^^^$ $ $d d d$ $ $ O 4
(( N$?Tf L
) ) ) ) )r2 r z^(powerpc|ppc)64zOnly for Linux and Powerc 8 e Zd Zg dZ eddd Zd ZdS )Test_POWER_Features)VSXVSX2VSX3VSX4 ARCH_2_07 ARCH_3_00ARCH_3_1)r r r c . | d S r6 ri r8 s r) r: z Test_POWER_Features.load_flagsl r2 N)rj rk rl rE r rK r: r7 r2 r) r r g sD ...H4[{LLLL r2 r z^(s390x)zOnly for Linux and IBM Zc e Zd Zg dZd ZdS )Test_ZARCH_Features)VXVXEVXE2c . | d S r6 r
r8 s r) r: z Test_ZARCH_Features.load_flagsv r r2 N)rj rk rl rE r: r7 r2 r) r
r
q s0 %$$H r2 r
z^(arm|aarch64)zOnly for Linux and ARMc > e Zd Zg dZ eddgddg Zd ZdS )Test_ARM_Features)NEONASIMDFPHPASIMDHPASIMDDPASIMDFHMr HALFVFPV4) NEON_FP16
NEON_VFPV4c V | d | d }|r0t dt t | z dk nd}t j dt s|rt ddd | _ d S t d
| _ d S ) NFeatureszCPU architecture0 r z^(aarch64|AARCH64)r )r r r )AESSHA1SHA2PMULLCRC32)r )
rR rP intnextiterr matchmachiner rK )r9 arch is_rootfs_v8s r) r: z Test_ARM_Features.load_flags s
+++$$%7888<Cs3tDJJ / //00144! H)7 3 3
|
$ 7'! ! !D !% @ ! ! !D r2 N)rj rk rl rE r rA r: r7 r2 r) r r { s\ H d f% g& O r2 r )!r platformr r numpy.core._multiarray_umathr r r numpynpr r r r* r1 r4 r r rq r\ is_linux is_cygwinr+ r*
IGNORECASEis_x86r is_powerr is_zarchr
is_armr r7 r2 r) r9 s_
-0 -0 -0^) ) )4 4 4 4 4 4 4 4l LL B A* A* A* A* A* A* A* A*F < " "7 + + L # #H - - 8 28,gr} E E Y -v:6N ) ) ) ) ) ) ) )> 28& ? ?L0L9STT , UT 28J 7 7L0L5 7 7 , 7 7
"GR] ; ;L.J7OPP QP r2