?????????? ????????? - ??????????????? - /usr/lib64/python3.9/site-packages/numpy/core/tests/__pycache__/test_simd.cpython-39.pyc
???????
a z[ycf� � @ s d dl Z d dlZd dlZd dlZd dlmZ d dlmZ G dd� d�ZG dd� de�Z G dd � d e�Z G d d � d e�Z G d d � d e�Z G dd� de�Z G dd� de�ZdZdZdZee Zee ee ee de de eeiZe�� D �]$\ZZer�ejndZe�d�Zee�dk�r"dd� e�� d �Zned ZdZ e � Z e�sHd e� d �Z n.ej�s^d e� d �Z nej �svd e� d �e d!< e�� D ]�\Z!Z"e!D ]tZ#e �$e#e �Z%e"fZ&e ee e#ed"�Z'e(d#e"j)� d$e� d$e� d$e#� �e&e'�Z*e%�r�e j+j e%d%�e*� e*e,� e*j)< �q��q~q�dS )&� N)�targets)�__cpu_baseline__c @ s� e Zd ZdZdZdZdd� Zd!dd�Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � ZdS )"� _Test_UtilityNc C s t | j|d | j �S )z� To call NPV intrinsics without the attribute 'npyv' and auto suffixing intrinsics according to class attribute 'sfx' �_��getattr�npyv�sfx)�self�attr� r �@/usr/lib64/python3.9/site-packages/numpy/core/tests/test_simd.py� __getattr__ s z_Test_Utility.__getattr__Fc C sR |du r d}|du r| j }t||| �}|r4t|�}| �� rJdd� |D �S t|�S )z[ Create list of consecutive numbers according to number of vector's lanes. N� c S s g | ] }|d �qS )g �?r ��.0�xr r r � <listcomp># � z'_Test_Utility._data.<locals>.<listcomp>)�nlanes�range�reversed�_is_fp�list)r �start�count�reverse�rngr r r �_data s z_Test_Utility._datac C s | j d dkS )Nr �u�r �r r r r � _is_unsigned& s z_Test_Utility._is_unsignedc C s | j d dkS )Nr �sr r! r r r � _is_signed) s z_Test_Utility._is_signedc C s | j d dkS )Nr �fr r! r r r r , s z_Test_Utility._is_fpc C s t | jdd � �S )Nr )�intr r! r r r � _scalar_size/ s z_Test_Utility._scalar_sizec s0 | � � r |S | �� � | �� �� �fdd�|D �S )Nc s g | ]}t t|��� ��qS r )�min�max)r �v�Zmax_intZmin_intr r r 7 r z+_Test_Utility._int_clip.<locals>.<listcomp>)r �_int_max�_int_min)r �seqr r+ r � _int_clip2 s z_Test_Utility._int_clipc C s4 | � � r d S | �| �d��d }| �� r0|d S |S )N���r � )r � _to_unsigned�setallr$ )r Zmax_ur r r r, 9 s z_Test_Utility._int_maxc C s& | � � r d S | �� rdS | �� d S )Nr r )r r" r, r! r r r r- A s z_Test_Utility._int_minc C s& t | jd| jdd � �d�}|d S )N�setall_ur r0 r r )r Z max_unsigr r r � _true_maskH s z_Test_Utility._true_maskc C st t |ttf�r,t| jd| jdd � �|�S |j�dd�}|d dkrLd}nd}t| j|�|dd � |��|�S d S ) N�load_ur �npyv_� r �bz cvt_u{0}_b{0}zreinterpret_u{0}_{1}) � isinstancer �tupler r r �__name__�replace�format)r �vectorr Z cvt_intrinr r r r2 L s z_Test_Utility._to_unsignedc C s | j �d�}| j �|�d S )Ni �r �r Z setall_u32Zreinterpret_f32_u32�r r* r r r � _pinfinityW s z_Test_Utility._pinfinityc C s | j �d�}| j �|�d S )Nl r r@ rA r r r � _ninfinity[ s z_Test_Utility._ninfinityc C s | j �d�}| j �|�d S )Ni �r r@ rA r r r �_nan_ s z_Test_Utility._nanc C s( | j }|dkrt}n |�d�}d�|�S )NZbaseline�__� )� target_namer �split�join)r �targetr r r � _cpu_featuresc s z_Test_Utility._cpu_features)NNF)r< � __module__� __qualname__r r rG r r r" r$ r r' r/ r, r- r5 r2 rB rC rD rK r r r r r s$ r c @ s2 e Zd ZdZd dd�Zdd� Zdd � Zd d � ZdS ) � _SIMD_BOOLz2 To test all boolean vector types at once NFc sH t | jd| jdd � �}| �� � t|�}|r6t|�}� fdd�|D �S )NZnlanes_ur c s g | ]}|d r� nd�qS )r1 r r r �� true_maskr r r u r z$_SIMD_BOOL._data.<locals>.<listcomp>)r r r r5 r r )r r r r r r r rO r r o s z_SIMD_BOOL._datac C sB | j dd � }t| jd| �}t| jd|� d|� ��}|||��S )Nr r6 Zcvt_bZ_u)r r r )r �dataZlen_str�loadZcvtr r r �_load_bw s z_SIMD_BOOL._load_bc C s� | � � }| j dd�}| �|�}| �|�}dd� t||�D �}t| d�||�}||ksXJ �dd� t||�D �}t| d�||�}||ks�J �dd� t||�D �} t| d �||�} | | ks�J �t| d �|�} | |ks�J �d S ) z� Logical operations for boolean types. Test intrinsics: npyv_xor_##SFX, npyv_and_##SFX, npyv_or_##SFX, npyv_not_##SFX T�r c S s g | ]\}}||@ �qS r r �r �ar9 r r r r � r z5_SIMD_BOOL.test_operators_logical.<locals>.<listcomp>�andc S s g | ]\}}||B �qS r r rU r r r r � r �orc S s g | ]\}}||A �qS r r rU r r r r � r �xor�notN)r rS �zipr ) r �data_a�data_b�vdata_a�vdata_b�data_and�vand�data_or�vor�data_xor�vxor�vnotr r r �test_operators_logical} s z!_SIMD_BOOL.test_operators_logicalc C sV dd� }| � � | j dd�fD ]4}| �|�}||�}t| �|��}|t|�ks J �q d S )Nc S s t dd� t| d�D ��S )Nc S s g | ]\}}t |d k�|> �qS )r �r& )r �ir r r r r � r z<_SIMD_BOOL.test_tobits.<locals>.<lambda>.<locals>.<listcomp>r )�sum� enumerate)rQ r r r �<lambda>� r z(_SIMD_BOOL.test_tobits.<locals>.<lambda>TrT )r rS �bin�tobits)r Z data2bitsrQ �vdataZ data_bitsrn r r r � test_tobits� s z_SIMD_BOOL.test_tobits)NNF)r< rL rM �__doc__r rS rg rp r r r r rN k s rN c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � _SIMD_INTz2 To test all integer vector types at once c s6 | j dv rd S | �| �� | j �}| j| �� dd�}| �|�| �|� }}t| �� �D ]d� | �� fdd�|D ��}| �|� �}||ks�J �| �� fdd�|D ��}| � |� �}||ksVJ �qVtd| �� �D ]f� | �� fdd�|D ��}| � |� �} | |k�s J �| �� fd d�|D ��}| � |� �} | |ks�J �q�d S ) N)�u8�s8TrT c s g | ] }|� > �qS r r �r rV �r r r r � r z2_SIMD_INT.test_operators_shift.<locals>.<listcomp>c s g | ] }|� ? �qS r r ru rv r r r � r r c s g | ] }|� > �qS r r ru rv r r r � r c s g | ] }|� ? �qS r r ru rv r r r � r ) r r r, r r- rR r r' �shl�shr�shli�shri) r r\ r] r^ r_ Z data_shl_arw Z data_shr_arx ry rz r rv r �test_operators_shift� s& z _SIMD_INT.test_operators_shiftc C s� | j dv rd S | �| �� | j �}| j| �� dd�}| �|�| �|� }}| �dd� t||�D ��}| �||�}||ks|J �| �dd� t||�D ��}| � ||�}||ks�J �d S )N)�u32�s32�u64�s64TrT c S s g | ]\}}|| �qS r r rU r r r r � r z>_SIMD_INT.test_arithmetic_subadd_saturated.<locals>.<listcomp>c S s g | ]\}}|| �qS r r rU r r r r � r ) r r r, r r- rR r/ r[ �adds�subs) r r\ r] r^ r_ Z data_addsr� Z data_subsr� r r r � test_arithmetic_subadd_saturated� s z*_SIMD_INT.test_arithmetic_subadd_saturatedc C s� | � � }| � | j�}| �|�| �|� }}dd� t||�D �}| �||�}||ksVJ �dd� t||�D �}| �||�}||ks�J �d S )Nc S s g | ]\}}t ||��qS r �r) rU r r r r � r z/_SIMD_INT.test_math_max_min.<locals>.<listcomp>c S s g | ]\}}t ||��qS r �r( rU r r r r � r )r r rR r[ r) r( ) r r\ r] r^ r_ �data_maxZsimd_max�data_minZsimd_minr r r �test_math_max_min� s z_SIMD_INT.test_math_max_minN)r< rL rM rq r{ r� r� r r r r rr � s !rr c @ s e Zd ZdZdd� ZdS )� _SIMD_FP32z' To only test single precision c sv | � � }| jjs&t�d|�r&dd� � nt� | �| �� �}| �|| � d��}� fdd�|D �}| � |�}||ksrJ �dS )�� Round to nearest even integer, assume CPU control register is set to rounding. Test intrinsics: npyv_round_s32_##SFX z.*(NEON|ASIMD)c S s t | | dkrdnd �S )Nr � �?� �rh )r* r r r rl � r z-_SIMD_FP32.test_conversions.<locals>.<lambda>r� c s g | ] }� |��qS r r r ��_roundr r r � r z/_SIMD_FP32.test_conversions.<locals>.<listcomp>N) rK r �simd_f64�re�match�roundrR r �subr3 � round_s32)r Zfeaturesr^ � data_round�vroundr r� r �test_conversions� s z_SIMD_FP32.test_conversionsN�r< rL rM rq r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )� _SIMD_FP64z' To only test double precision c C sh | � | �� �}| �|| �d��}| �|| �d��}dd� t|�t|� D �}| �||�}||ksdJ �dS )r� r� g ��c S s g | ] }t |��qS r )r� r r r r r r z/_SIMD_FP64.test_conversions.<locals>.<listcomp>N)rR r r� r3 �mulr r� )r r^ r_ r� r� r r r r� � s z_SIMD_FP64.test_conversionsNr� r r r r r� � s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zej � d d e j fd e j fd e jfdefg�dd� �Zdd� Zdd� Zdd� Zdd� ZdS )�_SIMD_FPz0 To test all float vector types at once c C s� | � | �� �gd \}}}| �||�}| � dd� t|||�D ��}| �|||�}||ks\J �| �|||�}| �||�}||ks�J �| �|||�} | �||�} | | ks�J �| �|||�} | � || � d��} | | ks�J �d S )N� c S s g | ]\}}}|| | �qS r r )r rV r9 �cr r r r r z2_SIMD_FP.test_arithmetic_fused.<locals>.<listcomp>r0 ) rR r �addr[ ZmuladdZmulsubr� ZnmuladdZnmulsubr� r3 ) r r^ r_ Zvdata_cZ vdata_cx2Zdata_fmaZfmaZfmsZdata_fmsZnfmaZ data_nfmaZnfmsZ data_nfmsr r r �test_arithmetic_fused s z _SIMD_FP.test_arithmetic_fusedc C s� | � � | �� | �� }}}| �� }| �| �� �}d||f||f||ff}|D ]:\}}|g| j } | �| �|��} | tj | dd�ksNJ �qN| �| � || �d���} | |ks�J �d S )N)r r T�Znan_okr0 ) rB rC rD r rR r �absr3 �pytest�approxr� ) r �pinf�ninf�nanrQ ro Z abs_cases�case�desiredZdata_absZvabsr r r �test_abs! s z_SIMD_FP.test_absc C s� | � � | �� | �� }}}| �� }| �| �� �}ddd|f||f||ff}|D ]:\}}|g| j } | �| �|��} | tj | dd�ksPJ �qP| �dd� |D ��} | �|�} | | ks�J �d S )N)� �r� )� r� g �Tr� c S s g | ]}t �|��qS r )�math�sqrtr r r r r : r z&_SIMD_FP.test_sqrt.<locals>.<listcomp>) rB rC rD r rR r r� r3 r� r� ) r r� r� r� rQ ro Z sqrt_casesr� r� Z data_sqrtr� r r r � test_sqrt/ s z_SIMD_FP.test_sqrtc C s� | � � | �� | �� }}}| �� }| �| �� �}||f||f||ff}|D ]:\}}|g| j } | �| �|��} | tj | dd�ksLJ �qLdd� |D �} | �|�} | | ks�J �d S )NTr� c S s g | ] }|| �qS r r r r r r r I r z(_SIMD_FP.test_square.<locals>.<listcomp>) rB rC rD r rR r �squarer3 r� r� ) r r� r� r� rQ ro Z square_casesr� r� Z data_squarer� r r r � test_square> s z_SIMD_FP.test_squarez intrin, func�ceil�trunc�floorZrintc s0 |}t | |�}| �� | �� | �� }}}||f||f||ff}|D ]8\}} | g| j } || �|��} | tj| dd�ksDJ �qDtddd�D ]T�dD ]J�| � ��fdd�t| j�D ��} � fd d�| D �} || �} | | ks�J �q�q�|d kr�d } nd } | D ]4�| � || �����} | � | �d ��} | | ks�J �q�dS )z� Test intrinsics: npyv_rint_##SFX npyv_ceil_##SFX npyv_trunc_##SFX npyv_floor##SFX Tr� r i � )g�������g�������gffffff�g�������?g�������?gffffff�?c s g | ]}�| � �qS r r ru )�wr r r r c r z*_SIMD_FP.test_rounding.<locals>.<listcomp>c s g | ] }� |��qS r r r )�funcr r r d r r� )r� )r� g пg333333ӿg������ܿr� r� N) r rB rC rD r r3 r� r� r rR r2 )r �intrinr� Z intrin_namer� r� r� Z round_casesr� r� r� r� rQ Z data_szeror )r� r� r r � test_roundingM s* z_SIMD_FP.test_roundingc C s\ | � � }| � | j�}| �|�| �|� }}dd� t||�D �}| �||�}||ksVJ �| �||�}||ksnJ �| �� | �� | �� }} } | | | f| ddfd| df|||f|d|fd||f| | | f| ddfd| dfddf } | D ]x\} } }|g| j }| � | �}| � | �}| �||�}|t j |dd�k�s,J �| | | |fv �r>q�| �||�}||ks�J �q�dS ) zY Test intrinsics: npyv_max_##SFX npyv_maxp_##SFX c S s g | ]\}}t ||��qS r r� rU r r r r | r z%_SIMD_FP.test_max.<locals>.<listcomp>� )r� r r� )r� ���r� Tr� N) r r rR r[ r) �maxprB rC rD r3 r� r� )r r\ r] r^ r_ r� Z_maxr� r� r� r� Z max_cases� case_operand1� case_operand2r� r r r �test_maxs s0 � z_SIMD_FP.test_maxc C s\ | � � }| � | j�}| �|�| �|� }}dd� t||�D �}| �||�}||ksVJ �| �||�}||ksnJ �| �� | �� | �� }} } | | | f| ddfd| df|||f|ddfd|df| | | f| d| fd| | fddf } | D ]x\} } }|g| j }| � | �}| � | �}| �||�}|t j |dd�k�s,J �| | | |fv �r>q�| �||�}||ks�J �q�dS ) zY Test intrinsics: npyv_min_##SFX npyv_minp_##SFX c S s g | ]\}}t ||��qS r r� rU r r r r � r z%_SIMD_FP.test_min.<locals>.<listcomp>r� )r� r r )r� r� r� Tr� N) r r rR r[ r( �minprB rC rD r3 r� r� )r r\ r] r^ r_ r� Z_minr� r� r� r� Z min_casesr� r� r� r r r �test_min� s0 � z_SIMD_FP.test_minc C s� | � � | �� | �� }}}| �� }| �| �� �}||f|df|dfd|fd|ff}|D ]:\}}|g| j } | �| �|��} | tj | dd�ksXJ �qX| �dd� |D ��} | �|�} | | ks�J �d S )Nr� r� Tr� c S s g | ] }d | �qS )r r r r r r r � r z,_SIMD_FP.test_reciprocal.<locals>.<listcomp>) rB rC rD r rR r �recipr3 r� r� ) r r� r� r� rQ ro Z recip_casesr� r� Z data_recipr� r r r �test_reciprocal� s " z_SIMD_FP.test_reciprocalc C s, | � | �| �� ��}|dg| j ks(J �dS )zQ Compare Not NaN. Test intrinsics: npyv_notnan_##SFX r N)Znotnanr3 rD r )r Znnanr r r �test_special_cases� s z_SIMD_FP.test_special_casesN)r< rL rM rq r� r� r� r� r� �mark� parametrizer� r� r� r� r� r� r� r� r� r� r r r r r� s � $ r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z e j �dddg�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3S )4� _SIMD_ALLz* To test all vector types at once c C s� | � � }| �|�}||ks J �| �|�}||ks4J �| �|�}||ksJJ �| �|�}t|�d | jd � }|d | jd � }||ks�J �||ks�J �d S )Nr1 )r rR Zloada�loads�loadlr r )r rQ Z load_dataZ loada_dataZ loads_datar� Z loadl_halfZ data_halfr r r �test_memory_load� s z_SIMD_ALL.test_memory_loadc C s | � � }| �|�}dg| j }| �||� ||ks6J �dg| j }| �||� ||ksZJ �dg| j }| �||� ||ks~J �dg| j }| �||� |d | jd � |d | jd � ks�J �||ks�J �dg| j }| �||� |d | jd � || jd d � k�s J �||k�sJ �d S )Nr r1 )r rR r �storeZstoreaZstoresZstorelZstoreh)r rQ ro r� Zstore_aZstore_sZstore_lZstore_hr r r �test_memory_store� s& ( *z_SIMD_ALL.test_memory_storec C s� | j dv rd S | �� }ttd| jd ��}|| jd | jd g7 }|D ]n}| �||d�}|d |� dg| j| }||ks�J �| �||�}|d |� dg| j| }||ksFJ �qFd S )N�rs rt �u16�s16r r1 � � r )r r r r r � load_till� load_tillz)r rQ �lanes�nr� � data_tillr� Z data_tillzr r r �test_memory_partial_load� s z"_SIMD_ALL.test_memory_partial_loadc C s� | j dv rd S | �� }| jdd�}| �|�}ttd| jd ��}|| jd | jd g7 }|D ]F}|�� }|d |� |d |�< | jdd�}| �|||� ||ks\J �q\d S )Nr� TrT r r1 r� )r r rR r r r �copy� store_till)r rQ Zdata_revro r� r� r� r� r r r �test_memory_partial_store s z#_SIMD_ALL.test_memory_partial_storec C s� | j dv rd S tdd�D ]<}| j|| j d�}|d d |� }| �||�}||ksJ �qtdd�D ]D}| �|| | j �}| �|d d |� �}| �||�}||ks`J �q`d S )Nr� r �@ rv ���r )r r r r �loadnrR )r �striderQ � data_strider� r r r �test_memory_noncont_load s z"_SIMD_ALL.test_memory_noncont_loadc C s� | j dv rd S ttd| jd ��}|| jd | jd g7 }tdd�D ]�}| j|| j d�}|d d |� }|D ]r}|d |� dg| j| }| �|||d�}||ks�J �|d |� dg| j| }| �|||�} | |kslJ �qlqDtd d�D ]�}| �|| | j �}t| �|d d |� ��}|D ]x}|d |� dg| j| }| �|||d�}||k�s^J �|d |� dg| j| }| �|||�} | |k�s J ��q q�d S ) Nr� r r1 r� r� rv r� r r� )r r r r r � loadn_till� loadn_tillzrR ) r r� r� rQ r� r� Zdata_stride_tillr� Zdata_stride_tillzr� r r r � test_memory_noncont_partial_load' s0 z*_SIMD_ALL.test_memory_noncont_partial_loadc C s6 | j dv rd S | �| �� �}tdd�D ]|}dg| | j }||d d |�< dg| | j }|dgd 7 }| �|||� |d d� |ks�J �|dd � dgd ks&J �q&tdd�D ]�}dg| | j }||d d |�< dgd }|dg| | j 7 }| �|||� |dd � |k�sJ �|d d� dgd ks�J �q�d S )Nr� r r� r� � r� r )r rR r r r �storen)r ro r� rQ r� r r r �test_memory_noncont_storeC s& z#_SIMD_ALL.test_memory_noncont_storec C s� | j dv rd S | �� }| �|�}ttd| jd ��}|| jd | jd g7 }tdd�D ]�}|D ]�}dg| | j }|d |� dg| j| |d d |�< dg| | j }|dgd 7 }| �||||� |d d� |ks�J �|dd � dgd ks^J �q^qVtdd �D ]�}|D ]�}dg| | j }|d |� dg| j| |d d |�< dgd }|dg| | j 7 }| �||||� |dd � |k�s�J �|d d� dgd k�s J ��q �qd S ) Nr� r r1 r� r� r� r� r� r )r r rR r r r � storen_till)r rQ ro r� r� r� r� r� r r r �!test_memory_noncont_partial_storeZ s0 & & z+_SIMD_ALL.test_memory_noncont_partial_storezintrin, table_size, elsize)z self.lut32� r� )z self.lut16� r� c C sj || � � krdS t|�}t| jd|� ��}td|�}|D ],}| �|�}||�}|||�} | |ks8J �q8dS )zi Test lookup table intrinsics: npyv_lut32_##sfx npyv_lut16_##sfx Nr4 r )r' �evalr r r r3 ) r r� Z table_sizeZelsizeZ idx_itrin�tableri Zbroadi�idxZlutr r r �test_lutv s z_SIMD_ALL.test_lutc C sX | � � }|dg| j ks J �tdd�D ]"}| �|�}||g| j ks&J �q&| �� | jdd� }}| �|�| �|� }}| j|� }||ks�J �| jdg|�R � } | |ks�J �g d�} | jj r�| � d� | D ](} t | d| �|�j } | d | ks�J �q�| � | �| � � | � � �||�} | |k�s J �| � | �| � � | � � �||�}||k�sJJ �| j�� d S ) Nr r r� TrT ) rs rt r� r� r| r} r~ r �f32�f64Z reinterpret_r7 )�zeror r r3 r rR �setZsetfr r� �appendr r< Zselect�cmpeq�cmpneqZcleanup)r Zbroadcast_zerori Z broadcastir\ r] r^ r_ ZvsetZvsetf�sfxesr Zvec_nameZselect_aZselect_br r r � test_misc� s, z_SIMD_ALL.test_miscc C s | � � | j dd� }}| �|�| �|� }}|d | jd � }|d | jd � }|| jd d � }|| jd d � }| �||�} | || ks�J �| �||�} | || ks�J �| �||�} | || || fks�J �dd� t||�D �} dd� t||�D �} | �||�}|| | fk�sJ �d S )NTrT r1 c S s g | ]}|D ]}|�q qS r r �r �pr* r r r r � r z*_SIMD_ALL.test_reorder.<locals>.<listcomp>c S s g | ]}|D ]}|�q qS r r r� r r r r � r )r rR r �combinel�combineh�combiner[ )r r\ r] r^ r_ Z data_a_loZ data_b_loZ data_a_hiZ data_b_hir� r� r� Z data_ziplZ data_ziphZvzipr r r � test_reorder� s z_SIMD_ALL.test_reorderc sZ | � � � � dkrd S � fdd�td| jd� �D �}| �| �t| j���}||ksVJ �d S )Nr� c s, g | ]$}t t||d � ��D ]}|�q qS )r� )r r )r r �y�Zssizer r r � s �z0_SIMD_ALL.test_reorder_rev64.<locals>.<listcomp>r )r' r r �rev64rR )r Z data_rev64r� r r� r �test_reorder_rev64� s �z _SIMD_ALL.test_reorder_rev64c s� | � � r| �� }n| �| �� | j �}| j| �� dd�}| �|�| �|� }}| �� � � fdd�}dd� t||�D �}|| �||��}||ks�J �dd� t||�D �}|| � ||��} | |ks�J �dd� t||�D �} || � ||��} | | ks�J �d d� t||�D �} || � ||��} | | k�s$J �d d� t||�D �}|| � ||��}||k�sVJ �d d� t||�D �}|| � ||��}||k�s�J �d S ) NTrT c s � fdd�| D �S )Nc s g | ] }|� k�qS r r )r Zlane�Z mask_truer r r � r zH_SIMD_ALL.test_operators_comparison.<locals>.to_bool.<locals>.<listcomp>r )r? r r r �to_bool� s z4_SIMD_ALL.test_operators_comparison.<locals>.to_boolc S s g | ]\}}||k�qS r r rU r r r r � r z7_SIMD_ALL.test_operators_comparison.<locals>.<listcomp>c S s g | ]\}}||k�qS r r rU r r r r � r c S s g | ]\}}||k�qS r r rU r r r r � r c S s g | ]\}}||k�qS r r rU r r r r � r c S s g | ]\}}||k �qS r r rU r r r r � r c S s g | ]\}}||k�qS r r rU r r r r � r )r r r, r r- rR r5 r[ r� r� �cmpgt�cmpge�cmplt�cmple)r r\ r] r^ r_ r Zdata_eqr� Zdata_neqr� Zdata_gtr Zdata_ger Zdata_ltr Zdata_ler r r r �test_operators_comparison� s2 z#_SIMD_ALL.test_operators_comparisonc C sr | � � r| �� }n| �| �� | j �}| j| �� dd�}| �|�| �|� }}| � � rz| �|�}| �|�}| j| j }}n|| }}dd� | j }}|dd� t||�D ��} || �||��} | | ks�J �|dd� t||�D ��} |t | d�||��} | | k�sJ �|d d� t||�D ��} |t | d �||��}|| k�s<J �|d d� |D ��}|t | d �|��}||k�snJ �d S ) NTrT c S s | S )Nr )rV r r r rl r z2_SIMD_ALL.test_operators_logical.<locals>.<lambda>c S s g | ]\}}||A �qS r r rU r r r r r z4_SIMD_ALL.test_operators_logical.<locals>.<listcomp>c S s g | ]\}}||B �qS r r rU r r r r r rX c S s g | ]\}}||@ �qS r r rU r r r r r rW c S s g | ] }| �qS r r ru r r r r r rZ ) r r r, r r- rR r2 r[ rY r )r r\ r] r^ r_ Z data_cast_aZ data_cast_b�castZ cast_datard re rb rc r` ra Zdata_notrf r r r rg � s. z _SIMD_ALL.test_operators_logicalc C s� d| j dd � }t| jd|| j f �}t| jd| j |f �}|| �d��}| �| �d�| �d��}||kspJ �||�}||�}||ks�J �d S )Nr9 r z cvt_%s_%sr )r r r r3 r� )r ZbsfxZ to_booleanZ from_booleanZfalse_vbZtrue_vbZ false_vsfxZ true_vsfxr r r �test_conversion_boolean s z!_SIMD_ALL.test_conversion_booleanc C s� | j dvrdS | j d tt| j dd� �d � }t| jd|� d| j � ��}| �| �� | j �}| �|�}||�}|d| jd � }|| jd d� }|||fks�J �dS )zh Test expand intrinsics: npyv_expand_u16_u8 npyv_expand_u32_u16 �rs r� Nr r r1 Zexpand_r ) r �strr& r r r r, r rR )r Ztotype�expandrQ ro ZedataZdata_loZdata_hir r r �test_conversion_expand( s $ z _SIMD_ALL.test_conversion_expandc C s� | � � r| �� }n| �| �� | j �}| j| �� dd�}| �|�| �|� }}| �dd� t||�D ��}| �||�}||ks�J �| �dd� t||�D ��}| �||�}||ks�J �d S )NTrT c S s g | ]\}}|| �qS r r rU r r r r E r z4_SIMD_ALL.test_arithmetic_subadd.<locals>.<listcomp>c S s g | ]\}}|| �qS r r rU r r r r H r ) r r r, r r- rR r[ r� r� ) r r\ r] r^ r_ �data_addr� �data_subr� r r r �test_arithmetic_subadd< s z _SIMD_ALL.test_arithmetic_subaddc C s� | j dv rd S | �� r | �� }n| �| �� | j �}| j| �� dd�}| �|�| �|� }}| �dd� t||�D ��}| �||�}||ks�J �d S )N)r~ r TrT c S s g | ]\}}|| �qS r r rU r r r r W r z1_SIMD_ALL.test_arithmetic_mul.<locals>.<listcomp>) r r r r, r r- rR r[ r� )r r\ r] r^ r_ Zdata_mulr� r r r �test_arithmetic_mulL s z _SIMD_ALL.test_arithmetic_mulc C sn | � � s d S | �� | jdd� }}| �|�| �|� }}| �dd� t||�D ��}| �||�}||ksjJ �d S )NTrT c S s g | ]\}}|| �qS r r rU r r r r c r z1_SIMD_ALL.test_arithmetic_div.<locals>.<listcomp>)r r rR r[ �div)r r\ r] r^ r_ Zdata_divr r r r �test_arithmetic_div[ s z _SIMD_ALL.test_arithmetic_divc sV | � � r dS | �� ��fdd��d� g}|tddd�7 }|tddd�7 }| �� }|d krz|tdd d �7 }|tdd d �7 }|d kr�|td dd�7 }|td dd�7 }|dkr�|tddd�7 }|tddd�7 }|dd� |D �7 }t�||�D ]h\}� | �� �d � � dk�r q�| �| �|��}� �fdd�|D �}| � � �}| � ||�}||ks�J �q�dS )zn Test integer division intrinsics: npyv_divisor_##sfx npyv_divc_##sfx Nc sR |dkr| � kr| S | dk |dk }}| dks6||kr>| | S | | | | d S )z� Divide towards zero works with large integers > 2^53, and wrap around overflow similar to what C does. r0 r r r )rV �dZsign_aZsign_d)�int_minr r � trunc_divq s z3_SIMD_ALL.test_arithmetic_intdiv.<locals>.trunc_divr r � r� � � r� i i� r� l i i��� l l l ���� c S s g | ] }| �qS r r r r r r r � r z4_SIMD_ALL.test_arithmetic_intdiv.<locals>.<listcomp>c s g | ]}�|� ��qS r r ru )�divisorr r r r � r ) r r- r r' � itertools�productr3 rR r r �divc)r rQ ZbsizeZdividendZ data_divcZ divisor_parmsr r )r r r r �test_arithmetic_intdivg s6 z _SIMD_ALL.test_arithmetic_intdivc C sB | j dvrdS | �� }| �|�}t|�}| �|�}||ks>J �dS )zH Test reduce sum intrinsics: npyv_sum_##sfx )r| r~ r� r� N)r r rR rj )r rQ ro �data_sum�vsumr r r �test_arithmetic_reduce_sum� s z$_SIMD_ALL.test_arithmetic_reduce_sumc C sj | j dvrdS d| j| �� | �� | j f}|D ]6}| �|�}| �|�}t|�}| �|�}||ks.J �q.dS )zQ Test extend reduce sum intrinsics: npyv_sumup_##sfx r Nr )r r r- r, r rR rj Zsumup)r Zrdata�rrQ ro r r r r r � test_arithmetic_reduce_sumup� s z&_SIMD_ALL.test_arithmetic_reduce_sumupc C s� | � | �� �}| � | jdd��}| �| �� | �� �}| �| �� | �� �}| �||�}| �||||�}||kspJ �| �||||�}||ks�J �| �||�}| �||||�}||ks�J �| �||||�}||ks�J �dS )z� Conditional addition and subtraction for all supported data types. Test intrinsics: npyv_ifadd_##SFX, npyv_ifsub_##SFX TrT N) rR r r� r� r� r� �ifsubr� �ifadd) r r^ r_ rP Z false_maskr r# r r$ r r r �test_mask_conditional� s z _SIMD_ALL.test_mask_conditionalN) r< rL rM rq r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r rg r r r r r r r r" r% r r r r r� � s8 � ( $ /r� )Zb8Zb16Zb32Zb64)rs rt r� r� r| r} r~ r )r� r� )r� )r� r8 rE r �(rF �)ztarget 'z$' isn't supported by current machinez' isn't supported by NPYVz"' doesn't support double-precisionr� )r r rG ZTestr )�reason)-r� r� r� r Znumpy.core._simdr Z numpy.core._multiarray_umathr r rN rr r� r� r� r� Zbool_sfxZint_sfxZfp_sfxZall_sfxZtests_registry�itemsrG r ZsimdZ simd_widthrH Z pretty_name�lenrI �skip�dictZskip_sfxr� r� �clsr �getZskip_mZinhrr �typer< Ztclsr� �globalsr r r r �<module> sb c4B @ � &
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????