U
|
O ±dº ã @ sL d dl Z d dlZd dlZd dlZd dlmZ dd ZG dd dZe j d¡Z e j d¡Z
|
e ¡ Z e d e ej ¡Zejje p~e
|
pe d
|
d G d d d eZe de ej ¡Zejje p¾e dd G dd deZe de ej ¡Zejje pöe dd G dd deZe de ej ¡Zejje p0e dd G dd deZdS )é N)Ú__cpu_features__c
|
C s d}t | t | } }| |kr"d S t t dd¡}z$tdd}| d¡}W 5 Q R X W n* tk
|
r } z t |}W 5 d }~X Y nX z*dd l}|jdgtd d
|
d } | ¡ } W n* tk
|
rÖ } z t |} W 5 d }~X Y nX dd l
|
}
|
|
|
j d ||| f d d} t d|| || f d S )NTú'Ú ú /proc/cpuinfoÚri r ú /bin/trueÚ1©Z LD_SHOW_AUXV©ÚenvaÄ
|
###########################################
|
### Extra debugging information
|
###########################################
|
-------------------------------------------
|
--- NumPy Detections
|
-------------------------------------------
|
%s
|
-------------------------------------------
|
--- SYS / CPUINFO
|
-------------------------------------------
|
%s....
|
-------------------------------------------
|
--- SYS / AUXV
|
-------------------------------------------
|
%s
|
ú )Úprefixz9Failure Detection
|
NAME: '%s'
|
ACTUAL: %s
|
DESIRED: %s
|
%s) Ústrr ÚreplaceÚopenÚreadÚ ExceptionÚ
|
subprocessÚ check_outputÚdictÚdecodeÚtextwrapÚindentÚAssertionError) ÚactualZdesiredÚfnameZ__tracebackhide__ZdetectedÚfdZcpuinfoÚerrr Úauxvr Z error_report© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_cpu_features.pyÚassert_features_equal s6 ðï
|
úr! c @ sN e Zd Zg Zi Zi Ze Zdd Zdd Z dd Z
|
dd Z d d
|
Z d d Z d S )Ú AbstractTestc C s d S ©Nr ©Úselfr r r Ú
|
load_flags9 s zAbstractTest.load_flagsc st ¡ j ¡ D ]0\}} fdd|D }tt |¡t|| q jD ]$} |¡}t |¡}t||| qJd S )Nc s g | ]} |¡qS r )Úcpu_have)Ú.0Úfr$ r r Ú
|
<listcomp>? s z.AbstractTest.test_features.<locals>.<listcomp>) r& Úfeatures_groupsÚitemsr! r ÚgetÚallÚfeaturesr' )r% Úgnamer/ Ú test_featuresÚ feature_namer' Znpy_haver r$ r r1 < s
|
|
|
zAbstractTest.test_featuresc C s@ | j ||¡}t|tr"|| jkS |D ]}|| jkr& dS q&dS )NTF)Ú features_mapr- Ú
|
isinstancer Úfeatures_flags)r% r2 Z map_namesr) r r r r' G s
|
|
|
zAbstractTest.cpu_havec C s | |¡| _d S r# )Úget_cpuinfo_itemr5 )r% Ú magic_keyr r r Úload_flags_cpuinfoP s z AbstractTest.load_flags_cpuinfoc C sl t }tdT}|D ]H}| |¡s$qdd | dd¡D }t|dkr| |d ¡ ¡ ¡}qW 5 Q R X |S )Nr c S s g | ] }| ¡ qS r ©Ústrip©r( Úsr r r r* Y s z1AbstractTest.get_cpuinfo_item.<locals>.<listcomp>ú:é é )Úsetr Ú
|
startswithÚsplitÚlenÚunionÚupper)r% r7 Úvaluesr ÚlineZ flags_valuer r r r6 S s
|
|
"z AbstractTest.get_cpuinfo_itemc C s~ dd l }|jdgtddd}| d¡D ]P}| d¡s8q(dd | d
|
d ¡D }t|d kr(| j |d ¡ ¡ ¡ ¡| _q(d S ) Nr r r r r
|
ó
|
s AT_HWCAPc S s g | ] }| ¡ qS r r9 r; r r r r* d s z0AbstractTest.load_flags_auxv.<locals>.<listcomp>ó :r> r? )
|
r r r rB rA rC r5 rD rE r )r% r r ÚatZ hwcap_valuer r r Úload_flags_auxv^ s
|
ÿz AbstractTest.load_flags_auxvN)Ú__name__Ú
|
__module__Ú __qualname__r/ r+ r3 r@ r5 r& r1 r' r8 r6 rK r r r r r" 3 s r" ÚlinuxÚcygwinz^(amd64|x86|i386|i686)zOnly for Linux and x86)Úreasonc @ sÒ e Zd Zddddddddd d
|
d d d ddddddddddddddd d g Zeddddgdddddddgdddddgddddddgdddddddgddddddddd d dg d Zed d d!d"d#d$d%d&d'd(d)
|
Zd*d+ Zd,S )-ÚTest_X86_FeaturesZMMXZSSEZSSE2ÚSSE3ZSSSE3ÚSSE41ZPOPCNTÚSSE42ZAVXZF16CZXOPZFMA4ÚFMA3ZAVX2ZAVX512FZAVX512CDZAVX512ERZAVX512PFÚ AVX5124FMAPSÚ AVX5124VNNIWÚAVX512VPOPCNTDQZAVX512VLZAVX512BWZAVX512DQÚ
|
AVX512VNNIZ
|
AVX512IFMAZ
|
AVX512VBMIÚ AVX512VBMI2Ú AVX512BITALG)Z
|
AVX512_KNLZ
|
AVX512_KNMZ
|
AVX512_SKXZ
|
AVX512_CLXZ
|
AVX512_CNLZ
|
AVX512_ICLZPNIZSSE4_1ZSSE4_2ZFMAZ AVX512_VNNIZ AVX512_BITALGZ AVX512_VBMI2Z AVX512_4FMAPSZ AVX512_4VNNIWZAVX512_VPOPCNTDQ)
|
rS rT rU rV rZ r\ r[ rW rX rY c C s | d¡ d S )NÚflags)r8 r$ r r r r& s z Test_X86_Features.load_flagsN)rL rM rN r/ r r+ r3 r& r r r r rR n s| û
|
|
ÿ ÿ ÿø ýrR z^(powerpc|ppc)64zOnly for Linux and Powerc @ s. e Zd ZddddgZeddddZd d
|
Zd S ) ÚTest_POWER_FeaturesZVSXÚVSX2ÚVSX3ÚVSX4Z ARCH_2_07Z ARCH_3_00ZARCH_3_1)r_ r` ra c C s | ¡ d S r# ©rK r$ r r r r& s z Test_POWER_Features.load_flagsN)rL rM rN r/ r r3 r& r r r r r^ s r^ z^(s390x)zOnly for Linux and IBM Zc @ s e Zd ZdddgZdd ZdS )ÚTest_ZARCH_FeaturesZVXZVXEZVXE2c C s | ¡ d S r# rb r$ r r r r& s z Test_ZARCH_Features.load_flagsN)rL rM rN r/ r& r r r r rc s
|
rc z^(arm|aarch64)zOnly for Linux and ARMc @ s8 e Zd ZddddddgZeddgddgd Zd
|
d Zd S ) ÚTest_ARM_FeaturesÚNEONÚASIMDZFPHPZASIMDHPZASIMDDPZASIMDFHMÚHALFÚVFPV4)Z NEON_FP16Z
|
NEON_VFPV4c C sf | d¡ | d¡}|r0tdtt| dknd}t dt¡sD|rVtdddd| _ n td d
|
| _ d S ) NZFeatureszCPU architectureÚ0é r z^(aarch64|AARCH64)rf )re rg rh )ZAESÚSHA1ZSHA2ZPMULLZCRC32)rf )
|
r8 r6 ÚintÚnextÚiterÚreÚmatchÚmachiner r3 )r% ÚarchZ is_rootfs_v8r r r r& ª s
|
|
ÿ
|
üz Test_ARM_Features.load_flagsN)rL rM rN r/ r r+ r& r r r r rd ¡ s ÿþrd )ÚsysÚplatformro ZpytestZ numpy.core._multiarray_umathr r! r" rA Zis_linuxZ is_cygwinrq rp Ú
|
IGNORECASEZis_x86ÚmarkZskipifrR Zis_powerr^ Zis_zarchrc Zis_armrd r r r r Ú<module> s. /7 ÿÿ
|
