
    sh1                         S SK r S SKJr  S SKJr  S SKJrJr  S SKJ	r	   " S S\5      r
\S:X  a  \R                  " 5         gg)	    N)cuda)driver)unittestContextResettingTestCase)Queuec                       \ rS rSrS rSrg)TestResetDevice   c                 F   S n[        5       n[        R                  " X4S9nUR                  5         UR	                  5         / nUR                  5       (       d6  UR                  UR                  5       5        UR                  5       (       d  M6  U R                  U/ 5        g )Nc                     [        [        R                  " 5       5      n[        S5       H7  nU H.  n[        R                  " U5        [        R
                  " 5         M0     M9     g ! [         a  nU R                  U5         S nAg S nAff = f)N   )ranger   get_device_countr   select_deviceclose	Exceptionput)exception_queuedevices_des        ~/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudadrv/test_reset_device.py	newthread4TestResetDevice.test_reset_device.<locals>.newthread   sl    ' 7 7 9:qA$**1-

 % "  '##A&&'s   A$A' '
B1BB)targetargs)	r   	threadingThreadstartjoinemptyappendgetassertEqual)selfr   r   t
exceptionss        r   test_reset_device!TestResetDevice.test_reset_device	   s    	'  'I4FG			
!''))o1134 "''))R(     N)__name__
__module____qualname____firstlineno__r)   __static_attributes__r,   r+   r   r	   r	      s    )r+   r	   __main__)r   numbar   numba.cuda.cudadrv.driverr   numba.cuda.testingr   r   queuer   r	   r-   mainr,   r+   r   <module>r8      s9      , A ). )6 zMMO r+   