Current File : //lib64/python3.9/site-packages/numpy/distutils/tests/__pycache__/test_ccompiler_opt.cpython-39.pyc
a
z[yc[p �
@ sH d dl Z d dlZd dlZd dlmZmZ d dlmZ edkoBedu Z 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
ddddddd�ZG dd� de�Z G dd� d�Z 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 s6 t j| i |��}z|V W t�|� n t�|� 0 d S �N)�tempfileZmkdtemp�shutil�rmtree)�args�kwargs�tmpdir� r �N/usr/lib64/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 )�x86�x64�ppc64�ppc64le�armhf�aarch64�s390xZnoarchc @ sD e Zd ZdZddd�Zdd� Zdd� Zdd � Zed
d �d d
��Z dS )�FakeCCompilerOptr c O s$ || _ || _tj| d fi |�� 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 |t�sJ �t |t�s J �| 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# �re�matchZ
dist_errorr$ �zip�join�len)r& Zsources�flagsr �src�fr r r � dist_compile8 s z FakeCCompilerOpt.dist_compilec C s t jS r )r" � fake_infor/ r r r � dist_infoF s zFakeCCompilerOpt.dist_infoF)�stderrc G s d S r r )r? r r r r �dist_logI s zFakeCCompilerOpt.dist_logN)r r )
�__name__�
__module__� __qualname__r= r% r0 r< r> � staticmethodr@ r r r r r" ' s
r" c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z d5dd�Z d6dd�Z
d7dd�Zi fdd�Zi fdd�Zdd � 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�_optr/ r r r � setup_classQ s z _Test_CCompilerOpt.setup_classc O s | j | jdft_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 )rF rK r/ r r r �optY s
z_Test_CCompilerOpt.optc C s
| � � jS r )rL r, r/ r r r �march^ s z_Test_CCompilerOpt.marchc C s
| � � jS r )rL r- r/ r r r r- a s z_Test_CCompilerOpt.cc_namec s� |t _| j|�dd�|�dd�|�dd�|�dd�d�}t� ��}tj�|d �}t|d
��}|� |� W d � n1 sx0 Y g � i }|�
|g�} | D ]t\}
} t� |
�� d �d d
� } t
| �}
|
dkr�d} n,|
d kr�| d �� } ndd�| � d �� } � �| � | || < q�W d � n1 �s&0 Y |j| \}}|�rN|dg n|}dd� |D �}t
|�t
� �k�s�t� fdd�|D ���s�td|� f ��||fS )N�baseline�min�dispatch�maxr' r r( �� cpu_baseline� cpu_dispatchr' r( ztest_targets.c�w�.� ���r �(r1 �)c S s, g | ]$}t |t�r$d d�|� d n|�qS )rY r1 rZ )r2 �tupler7 )�.0�tarr r r �
<listcomp>� s �z2_Test_CCompilerOpt.get_targets.<locals>.<listcomp>c 3 s | ]}|� v V qd S r r )r\ �t�Zgtargetsr r � <genexpr>� � z1_Test_CCompilerOpt.get_targets.<locals>.<genexpr>zM'sources_status' returns different targets than the compiled targets
%s != %s)r" Zconf_target_groupsrK �getr �osr r7 �open�writeZ try_dispatch�basename�splitr8 �upper�appendZsources_status�allr )r& �targets�groupsr rL r
�filer; ZgflagsZ fake_objects�sourcer9 ZgtarZglenZ has_baseliner r` r � get_targetsd sH
� (
*�*��z _Test_CCompilerOpt.get_targetsc K s� t ddddd�}| �� }| �� }|�||�}|�||�}|||||d | |d | |d | |d | fD ] }|�|d �}|d url q�ql|r�t|t �r�|�� D ]$\} }
|
dd � dvr�|
d || < q�n&t|t�s�J �|dd � dvr�|d 7 }|S )
Nr r r r )r r r r �_rX z)}$?\.+*�$)�dictrM r- rc �popr2 �items�str) r& r Z
map2originrM r- Z map_marchZmap_cc�keyZregex�k�vr r r � arg_regex� s6 �
�
z _Test_CCompilerOpt.arg_regexr c K s� | j f i |��}|d u r d S | j|||�dd�|�dd�d�}d�|�� �}|slt|�dkrhtd| ��d S t�||tj �s�td||f ��d S ) Nr' r r( rR r1 r �!expected empty features, not "%s"z%dispatch features "%s" not match "%s")
rz rK rc r7 �cpu_dispatch_namesr8 r r4 r5 �
IGNORECASE)r& rP rN r r5 rL �featuresr r r �expect� s&
� �
�z_Test_CCompilerOpt.expectc K s� | j f i |��}|d u r d S | j|||�dd�|�dd�d�}d�|�� �}|slt|�dkrhtd| ��d S t�||tj �s�td||f ��d S ) Nr' r r( rR r1 r r{ z%baseline features "%s" not match "%s")
rz rK rc r7 �cpu_baseline_namesr8 r r4 r5 r} )r& rN rP r r5 rL r~ r r r �expect_baseline� s&
� �
�z"_Test_CCompilerOpt.expect_baselinec K s� | j f i |��}|d u r d S | j|||�dd�|�dd�d�}d�|�� �}|slt|�dkrhtd| ��d S t�||�s�td||f ��d S ) Nr' r r( rR r1 r z expected empty flags not "%s"zflags "%s" not match "%s") rz rK rc r7 Zcpu_baseline_flagsr8 r r4 r5 )r& rN rP r r5 rL r9 r r r � expect_flags� s&
� �
�z _Test_CCompilerOpt.expect_flagsc K s� | j f i |��}|d u r d S | jf ||d�|��\}}d�|�}|s`t|�dkr\td| ��d S t�||tj�s�td||f ��d S )N�rl rm r1 r z expected empty targets, not "%s"ztargets "%s" not match "%s")rz rp r7 r8 r r4 r5 r} )r& rl rm r r5 rq 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 ]^\}}||vr`td| ��|| } |s�t| �dkr�td| ��t�|| �sDtd|| |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") rz r2 rs rp ru r r8 r4 r5 )
r& rl rm r Z
match_dictrq Z tar_flagsZ match_tarZ match_flagsr9 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_)rI rJ rL r �getattr)r& Z
wrong_archZwrong_ccrL r r r �test_interface s z!_Test_CCompilerOpt.test_interfacec C sH dD ]>\}}| j ||d�}t|�� �dks.J �t|�� �dksJ �qd S )N))r ZnonerH )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�rS rT r )rK r8 r� r| )r& rN rP rL r r r �test_args_empty s z"_Test_CCompilerOpt.test_args_emptyc C sP | � � dkrd S dD ]6\}}z| j||d� td��W q tyH Y q0 qd S )N�unknown))zunkown_feature - max +minzunknown max min)z#avx2z$vsxr� z+excepted an exception for invalid arguments)rM rK r r )r& rN rP 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)rN r r r zavx2 vsx3 asimdhp)r r/ r r r � test_skip; s& ����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� q�d 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.*%sr1 )�x86_msvcr� �vsx vsx2)r )r� Z neon_fp16Z
neon_vfpv4�asimd)r )r� r7 )r& Zf0�f1�diffr; Zffr r r � test_impliesS s0 � �� �
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* t�y | �� d"k�rtd$��Y n0 d S )%N)rQ �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! rO 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 rM 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� rM r/ r r r �
test_flags� sF � ������z _Test_CCompilerOpt.test_flagsc C sj dD ]`}z4| j |ddddd� | �� dkr:td| �� ��W q tyb | �� dkr^td��Y q0 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� rM r r �r& rl 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)rN 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)rN rP 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)rN 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) rN r� r� r� r� r r r r r� r/ r r r � test_targets s< ��
��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)rN 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 */rN z.*-O3.*z.*/O3.*z.*/O2.*r� )r r r r r� z /*@targets baseline $werror */z .*-Werror.*z .*/Werror.*z.*/WX.*)r� rM r� )r& Zpolicyr r r �test_targets_policies] sT � �
�� � �z(_Test_CCompilerOpt.test_targets_policiesc C s> | j dtdd�dddd� | 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)rm 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)rm r r r r r )r� rs r/ 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� r/ r r r �test_targets_multi� s2 �
���z%_Test_CCompilerOpt.test_targets_multi)r )r )r ) rA rB rC rI rJ rG rK rL rM r- rp rz r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rE M s6 ,
)8' $@L,rE 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()
rq )�
class_namerI rJ zo class TestCCompilerOpt_{class_name}(_Test_CCompilerOpt):
arch = '{arch}'
cc = '{cc}'
)�
is_standaloner) r* r+ )rI rJ r r r �new_test s �
�r� )r )%r4 r) rd r r Zdistutils.errorsr rA � __package__r� Zunittest�
contextlibr r rj �abspathr7 �dirname�__file__Z
ccompiler_optr �contextmanagerr r Z numpy.distutils.ccompiler_optZ
numpy.testingrs Zarch_compilersr" rE r� ru rI Z compilersrJ �exec�mainr r r r �<module> sF
� & @
Mini Shell