o
    i                     @   sT   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r(e
  dS 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   	to_devicecopy_to_devicecopy_to_hostsynchronize
assertTrueallclose)selfAA0r   ptr r   b/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/cuda/tests/cudadrv/test_pinned.py_run_copies
   s   
zTestPinned._run_copiesc                 C   sj   t  }|ds|drd}nd}t|}t| | | W d    d S 1 s.w   Y  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   numbar   numba.cuda.testingr   r   r   r!   mainr   r   r   r   <module>   s    