
    J/Ph                     |    d dl Zd dlZd dlmZ d dlmZmZ  G d de          Ze	dk    r ej
                     dS dS )    N)cuda)unittestContextResettingTestCasec                        e Zd Zd Zd Zd ZdS )
TestPinnedc                 V   t          j        |          }t          j                    }t          j        |d|          }|                    ||           |                    ||           |                                 |                     t          j	        ||                     d S )NF)copystream)r
   )
npr	   r   r
   	to_devicecopy_to_devicecopy_to_hostsynchronize
assertTrueallclose)selfAA0r
   ptrs        d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/cuda/tests/cudadrv/test_pinned.py_run_copieszTestPinned._run_copies
   s    WQZZnQU6:::1V,,,6***Ar**+++++    c                 6   t          j                    }|                    d          s|                    d          rd}nd}t          j        |          }t          j        |          5  |                     |           d d d            d S # 1 swxY w Y   d S )Narmaarch64i       )platformmachine
startswithr   aranger   pinnedr   )r   r   countr   s       r   test_pinnedzTestPinned.test_pinned   s    "$$e$$ 	(:(:9(E(E 	EEEIe[^^ 	  	 Q	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	 s   +BBBc                 X    t          j        d          }|                     |           d S )Nr   )r   r    r   )r   r   s     r   test_unpinnedzTestPinned.test_unpinned   s+    Io&&r   N)__name__
__module____qualname__r   r#   r%    r   r   r   r      sA        	, 	, 	,         r   r   __main__)numpyr   r   numbar   numba.cuda.testingr   r   r   r&   mainr)   r   r   <module>r/      s               A A A A A A A A    )   8 zHMOOOOO r   