U
    -e                     @   sP   d dl Zd dlZd dlmZ d dlmZmZ G dd deZe	dkrLe
  dS )    N)cuda)unittestContextResettingTestCasec                   @   s$   e Zd Zdd Zdd Zdd ZdS )
TestPinnedc                 C   s\   t |}t }tj|d|d}|j||d |j||d |  | t 	|| d S )NF)copystream)r   )
npr   r   r   Z	to_deviceZcopy_to_deviceZcopy_to_hostZsynchronize
assertTrueZallclose)selfAZA0r   Zptr r   e/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/numba/cuda/tests/cudadrv/test_pinned.py_run_copies
   s    
zTestPinned._run_copiesc              	   C   sT   t  }|ds|dr"d}nd}t|}t| | | W 5 Q R X d S )Narmaarch64i       )platformmachine
startswithr   aranger   pinnedr   )r
   r   countr   r   r   r   test_pinned   s    
zTestPinned.test_pinnedc                 C   s   t d}| | d S )Nr   )r   r   r   )r
   r   r   r   r   test_unpinned   s    
zTestPinned.test_unpinnedN)__name__
__module____qualname__r   r   r   r   r   r   r   r      s   
r   __main__)numpyr   r   Znumbar   Znumba.cuda.testingr   r   r   r   mainr   r   r   r   <module>   s   