File: //lib/python3/dist-packages/numpy/distutils/tests/__pycache__/test_ccompiler_opt.cpython-310.pyc
o
6��a�l � @ sF d dl Z d dlZd dlZd dlmZmZ d dlmZ edko!edu Z e rWd dl
Z
d dl Z d dl Z d dl
Z
ej�e�e�e�e�d��� d dlmZ e jdd� �Zdd
d �Znd dlmZ d d lmZmZ ed
d
dddddd�ZG dd� de�Z G dd� d�Z dd� Z e� � D ]\Z Z!e!D ] Z"e#e e e"�� q�q�e r�e
�$� dS dS )� N)�sys�path)�DistutilsError�__main__z..)� CCompilerOptc o s6 � t j| i |��}z |V W t�|� d S t�|� w �N)�tempfile�mkdtemp�shutil�rmtree)�args�kwargs�tmpdir� r �J/usr/lib/python3/dist-packages/numpy/distutils/tests/test_ccompiler_opt.py�tempdir s
�r � c C s | st |��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�noarchc @ 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�sJ �| jr"|D ]}t�| j|�r!| �d� q| jr6|D ]}t�| j|�r5| �d� q't|d�|�gt |� �S )Nz%source is trapped by a fake interfacez#flag is trapped by a fake interface� )
�
isinstance�listr$ �re�match�
dist_errorr% �zip�join�len)r' �sources�flagsr
�src�fr r r � dist_compile7 s
�
�z FakeCCompilerOpt.dist_compilec C s t jS r )r# � fake_infor0 r r r � dist_infoE s zFakeCCompilerOpt.dist_infoF)�stderrc G s d S r r )rB r r r r �dist_logH s zFakeCCompilerOpt.dist_logN)r r )
�__name__�
__module__� __qualname__r@ r& r1 r? rA � staticmethodrC 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 d5dd�Z
d5dd�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 )6�_Test_CCompilerOptNc C s dt _d | _d S )NT)r# � conf_nocache�_optr0 r r r �setupP s
z_Test_CCompilerOpt.setupc O s | j | jdft_t|i |��S �Nr )�arch�ccr# r@ )r' r r
r r r �noptT s z_Test_CCompilerOpt.noptc C s | j s| �� | _ | j S r )rJ rO r0 r r r �optX s
z_Test_CCompilerOpt.optc C �
| � � jS r )rP r- r0 r r r �march] �
z_Test_CCompilerOpt.marchc C rQ r )rP r. r0 r r r r. ` rS z_Test_CCompilerOpt.cc_namec s� |t _| j|�dd�|�dd�|�dd�|�dd�d�}t� �p}tj�|d �}t|d
��
}|� |� W d � n1 s<w Y g � i }|�
|g�} | D ]:\}
} t� |
�� d �d d
� } t
| �}
|
dkrhd} n|
d krs| d �� } n dd�| � d �� } � �| � | || < qMW d � n1 s�w Y |j| \}}|r�|dg n|}dd� |D �}t
|�t
� �ks�t� fdd�|D ��s�td|� 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 |t�rd d�|� d n|�qS )r_ r2 r` )r3 �tupler9 )�.0�tarr r r �
<listcomp>� s ��z2_Test_CCompilerOpt.get_targets.<locals>.<listcomp>c 3 s � | ]}|� v V qd S r r )rb �t��gtargetsr r � <genexpr>� s � z1_Test_CCompilerOpt.get_targets.<locals>.<genexpr>zM'sources_status' returns different targets than the compiled targets
%s != %s)r# �conf_target_groupsrO �getr �osr r9 �open�write� try_dispatch�basename�splitr: �upper�append�sources_status�allr )r' �targets�groupsr
rP r �filer> �gflags� fake_objects�sourcer< �gtar�glen� has_baseliner rf r � get_targetsc sN
� �
� ��&��z _Test_CCompilerOpt.get_targetsc K s� t ddddd�}| �� }| �� }|�||�}|�||�}|||||d | |d | |d | |d | fD ]}|�|d �}|d urD nq6|rxt|t �re|�� D ]\} }
|
dd � dvrb|
d || < qP|S t|t�slJ �|dd � dvrx|d 7 }|S )
Nr r r r )r r r! r �_r^ z)}$?\.+*�$)�dictrR r. rj �popr3 �items�str) r' r
�
map2originrR r. � map_march�map_cc�key�regex�k�vr r r � arg_regex� s< �
� �
��z _Test_CCompilerOpt.arg_regexr c K s� | j d i |��}|d u rd S | j|||�dd�|�dd�d�}d�|�� �}|s6t|�dkr4td| ��d S t�||tj �sFtd||f ��d S )
Nr( r r) rX r2 r �!expected empty features, not "%s"z%dispatch features "%s" not match "%s"r )
r� rO rj r9 �cpu_dispatch_namesr: r r5 r6 �
IGNORECASE)r' rV rT r
r6 rP �featuresr r r �expect� �(
� �
��z_Test_CCompilerOpt.expectc K s� | j d i |��}|d u rd S | j|||�dd�|�dd�d�}d�|�� �}|s6t|�dkr4td| ��d S t�||tj �sFtd||f ��d S )
Nr( r r) rX r2 r r� z%baseline features "%s" not match "%s"r )
r� rO rj r9 �cpu_baseline_namesr: r r5 r6 r� )r' rT rV r
r6 rP r� r r r �expect_baseline� r� z"_Test_CCompilerOpt.expect_baselinec K s� | j d i |��}|d u rd S | j|||�dd�|�dd�d�}d�|�� �}|s6t|�dkr4td| ��d S t�||�sDtd||f ��d S )
Nr( r r) rX r2 r z expected empty flags not "%s"zflags "%s" not match "%s"r ) r� rO rj r9 �cpu_baseline_flagsr: r r5 r6 )r' rT rV r
r6 rP r< r r r � expect_flags� s(
� �
��z _Test_CCompilerOpt.expect_flagsc K s� | j di |��}|d u rd S | jd||d�|��\}}d�|�}|s0t|�dkr.td| ��d S t�||tj�s@td||f ��d S )N�ru rv r2 r z expected empty targets, not "%s"ztargets "%s" not match "%s"r )r� r~ r9 r: r r5 r6 r� )r' ru rv r
r6 r r r r �expect_targets� s
�
��z!_Test_CCompilerOpt.expect_targetsc
K s� | j di |��}|d u rd S t|t�sJ �| j||d�\}}|�� D ]/\}}||vr0td| ��|| } |sBt| �dkrBtd| ��t�|| �sQtd|| |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 ) r� r3 r� r~ r� r r: r5 r6 )
r' ru rv r
�
match_dictr � tar_flags� match_tar� match_flagsr< r r r �expect_target_flags� s, � � ���z&_Test_CCompilerOpt.expect_target_flagsc C s� | j dkrdnd}| jdkrdnd}| �� }tt|d| j �� tt|d| � � tt|d| j �� tt|d| � � d S )Nr r r r �cc_on_�cc_is_)rM rN rP r �getattr)r' �
wrong_arch�wrong_ccrP r r r �test_interface s z!_Test_CCompilerOpt.test_interfacec C sH dD ] \}}| j ||d�}t|�� �dksJ �t|�� �dks!J �qd S )N))r �nonerL )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�rY rZ r )rO r: r� r� )r' rT rV rP r r r �test_args_empty s
�z"_Test_CCompilerOpt.test_args_emptyc C sJ | � � dkrd S dD ]\}}z | j||d� td�� ty" Y q
w d S )N�unknown))zunkown_feature - max +minzunknown max min)z#avx2z$vsxr� z+excepted an exception for invalid arguments)rR rO r r )r' rT rV 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)rT r r r zavx2 vsx3 asimdhp)r� r0 r r r � test_skip: s& ���
�z _Test_CCompilerOpt.test_skipc C s� | j ddddd� dD ] \}}d||f }| j |d| ||d � | j |d
| ||d � q d D ]}|D ]}| j |d d
�|� d� q3q/| j ddd� dD ] }| j |dd� qLd S )Nzfma3 avx2 asimd vsx3zsse .* sse41 .* fma3.*avx2z
vsx vsx2 vsx3z neon neon_fp16 neon_vfpv4 asimd�r r r )��fma3�avx2)�avx512f�avx512cdz.* sse42 .* %s .*%s$z.* sse42 .* %s$)�x86_gcc�x86_icc�x86_iccwz
.* avx .* %s$)r� )r� r� �
avx512_skxz.*%sr2 )�x86_msvcr� �vsx vsx2)r )r� � neon_fp16�
neon_vfpv4�asimd)r! )r� r9 )r' �f0�f1�diffr> �ffr r r � test_impliesR s6 � �� ���z _Test_CCompilerOpt.test_impliesc
C s dD ]9}|dkr| � � dkrq| j|ddddd� | j|ddd d
ddd � | j|d d
ddd� | j|ddddddd� q| jdddddddd� | jddddd� z| jdd d d d d � | �� d krjtd | �� ��W d S ty� | �� d kr}td!��Y d S w )"N)rW �nativer� r z.*cpu_(sse|vsx|neon).cr )r( r r r z .*cpu_(sse3|vsx2|neon_vfpv4).czsse sse2r� zneon neon_fp16)r( r r r r! r z.*cpu_(popcnt|vsx3).cz sse .* sse41r� zneon neon_fp16 .* asimd .*z=.* 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.*)r� r� r� r� r r rU z
sse sse2 sse3zneon neon_fp16 .* asimd)r r r r! r r z.*cpu_(sse2|vsx2).c)r( r r z".*(-march=native|-xHost|/QxHost).*�.*)r) r r r r� � excepted an exception for %s�excepted no exceptions)r. r� rR r r )r' �or r r �test_args_options{ sb ���� ���
�� ��z$_Test_CCompilerOpt.test_args_optionsc C sj | j dddddddddd� | j d dd
d � | j d d
dd� | j ddd� | j ddd� | j ddd� d S )Nz sse sse2 vsx vsx2 neon neon_fp16z -msse -msse2z
/arch:SSE2z -mcpu=power8z -maltivec -mvsx -mpower8-vectorz"-mfpu=neon-fp16 -mfp16-format=ieeer )r� r� r� r� � ppc64_gcc� ppc64_clang� armhf_gccr! 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\+fp16fml)r� r0 r r r �
test_flags� s: �
����
�z _Test_CCompilerOpt.test_flagsc C sf dD ].}z| j |dddd� | �� dkr td| �� ��W q ty0 | �� dkr.td��Y qw d 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� rR r r �r' ru r r r �test_targets_exceptions� s" �
�� ���z*_Test_CCompilerOpt.test_targets_exceptionsc C s" dD ] }| j |ddddd� qd S )N)z(/*@targets $keep_baseline sse vsx neon*/z(/*@targets,$keep_baseline,sse,vsx,neon*/z(/*@targets*$keep_baseline*sse*vsx*neon*/zo
/*
** @targets
** $keep_baseline, sse vsx,neon
*/
z�
/*
************@targets*************
** $keep_baseline, sse vsx, neon
*********************************
*/
z�
/*
/////////////@targets/////////////////
//$keep_baseline//sse//vsx//neon
/////////////////////////////////////
*/
zg
/*
@targets
$keep_baseline
SSE VSX NEON*/
r� r� r� r r� �r� r� r r r �test_targets_syntax� s
��z&_Test_CCompilerOpt.test_targets_syntaxc C sb | j dddddd� | j ddd d
d d d
� | j ddddd dd� | j ddddddddddd�
d S )Nz�
/*@targets
sse sse2 sse41 avx avx2 avx512f
vsx vsx2 vsx3
neon neon_fp16 asimdhp asimddp
*/
zavx vsx2 asimdz avx512f avx2zasimddp asimdhp�vsx3)rT r r r z�
/*@targets
sse41 avx avx2 avx512f
vsx2 vsx3
asimd asimdhp asimddp
*/
r z sse41 avx2 vsx2 asimd asimddpz
avx2 sse41�
asimddp asimd�vsx2)rT rV r r r z�
/*@targets
sse2 sse41 avx2 avx512f
vsx2 vsx3
neon asimdhp asimddp
*/
z .*(avx2|avx512f|vsx3|asimddp).cz
sse41 sse2z asimdhp neon)rT 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) rT r� r� r� r� r r r r! r� r0 r r r � test_targets s8 � � �
�z _Test_CCompilerOpt.test_targetsc
C s� | j ddddddd� | j dd d
d d d
� | j ddddddddd� dD ]}| jd| ddiddiddiddiddid� q$| jdddiddidd idd iddid� d S ) Nz�
/*@targets
$keep_baseline
sse2 sse42 avx2 avx512f
vsx2 vsx3
neon neon_vfpv4 asimd asimddp
*/
zsse41 avx2 vsx2 asimd vsx3zavx512f avx2 sse42 sse2r� z asimddp asimd neon_vfpv4 neonr� )rT r r r r! z�
/*@targets
$keep_baseline $keep_sort
avx512f sse42 avx2 sse2
vsx2 vsx3
asimd neon neon_vfpv4 asimddp
*/
zavx512f sse42 avx2 sse2r� z asimd neon neon_vfpv4 asimddpz
asimd asimddp)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 sse2zavx512f avx2 sse2)r� r� r� r� r r r! )z$maxoptz$autovecz/*@targets baseline %s */rT z.*-O3.*z.*/O3.*z.*/O2.*r� )r r r r r� z /*@targets baseline $werror */z .*-Werror.*z .*/Werror.*z.*/WX.*)r� r� )r' �policyr r r �test_targets_policiesO sL ��� �
�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
)�
test_groupzavx512f avx2 sse2 baselinezvsx3 vsx2 baselinezasimddp asimdhp baseline)rv 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� r� zasimdfhm asimddp asimdzasimdfhm asimddp)rv r r r r r! )r� r� r0 r r r �test_targets_groups� s* �� �
�z&_Test_CCompilerOpt.test_targets_groupsc C sN | j dddd� | j dddddd � | j d
d ddd � | j d
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)
*/
z avx avx512fr� 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� s0 �
��
�z%_Test_CCompilerOpt.test_targets_multi�r ) rD rE rF rM rN rK rO rP rR 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 rH L s8 ,
)6! $;G ,rH c C s@ t rt�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()
r )�
class_namerM rN zo class TestCCompilerOpt_{class_name}(_Test_CCompilerOpt):
arch = '{arch}'
cc = '{cc}'
)�
is_standaloner* r+ r, )rM rN r r r �new_test� s �
�r� r� )%r5 r* rk r r �distutils.errorsr rD � __package__r� �unittest�
contextlibr r
rr �abspathr9 �dirname�__file__�
ccompiler_optr �contextmanagerr r � numpy.distutils.ccompiler_opt�
numpy.testingr� �arch_compilersr# rH r� r� rM � compilersrN �exec�mainr r r r �<module> sL
�
& , � �