
    i                     v    S SK rS SKrS SKJr  S SKJrJr   " S S\5      r\	S:X  a  \R                  " 5         gg)    N)cuda)unittestContextResettingTestCasec                   &    \ rS rSrS rS rS rSrg)
TestPinned   c                 ,   [         R                  " U5      n[        R                  " 5       n[        R                  " USUS9nUR                  XS9  UR                  XS9  UR                  5         U R                  [         R                  " X5      5        g )NF)copystream)r   )
npr
   r   r   	to_devicecopy_to_devicecopy_to_hostsynchronize
assertTrueallclose)selfAA0r   ptrs        _/var/www/html/trading/venv/lib/python3.13/site-packages/numba/cuda/tests/cudadrv/test_pinned.py_run_copiesTestPinned._run_copies
   so    WWQZnnQU6:1,*A*+    c                 >   [         R                  " 5       nUR                  S5      (       d  UR                  S5      (       a  SnOSn[        R                  " U5      n[
        R                  " U5         U R                  U5        S S S 5        g ! , (       d  f       g = f)Narmaarch64i       )platformmachine
startswithr   aranger   pinnedr   )r   r    countr   s       r   test_pinnedTestPinned.test_pinned   sl    ""$e$$(:(:9(E(EEEIIe[[^Q ^^s   3B
Bc                 R    [         R                  " S5      nU R                  U5        g )Nr   )r   r"   r   )r   r   s     r   test_unpinnedTestPinned.test_unpinned   s    IIo&r    N)__name__
__module____qualname____firstlineno__r   r%   r(   __static_attributes__r*   r   r   r   r      s    	, r   r   __main__)numpyr   r   numbar   numba.cuda.testingr   r   r   r+   mainr*   r   r   <module>r5      s6       A) 8 zMMO r   