o
    ãÊiÄ  ã                   @   sN   d dl Z d dlZe  ej ¡  d¡G dd„ de jƒƒZedkr%e  ¡  dS dS )é    NzCUDA not availablec                   @   s   e Zd Zdd„ ZdS )ÚTestLeanImportc                 C   s,   dd„ }t jj}|t j_dd l}|t j_d S )Nc                   S   s   t dƒ‚)Nzdo not call me)ÚAssertionError© r   r   úI/home/ubuntu/.local/lib/python3.10/site-packages/test/test_lean_import.pyÚ_patched_current_device   s   z\TestLeanImport.test_torchao_import_does_not_initialize_cuda.<locals>._patched_current_devicer   )ÚtorchÚcudaÚcurrent_deviceÚtorchao)Úselfr   Úold_current_devicer
   r   r   r   Ú,test_torchao_import_does_not_initialize_cuda   s
   z;TestLeanImport.test_torchao_import_does_not_initialize_cudaN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s    r   Ú__main__)	Úunittestr   ÚskipIfr   Úis_availableÚTestCaser   r   Úmainr   r   r   r   Ú<module>   s    ÿ