o
    ÔÙ¾iC  ã                   @   sN   d dl Z d dlmZ d dlmZmZ e  e¡Zdd„ Z	e	ƒ Z
e
o"eƒ ZeZdS )é    N)Úenvs)Úget_device_smÚis_blackwell_supportedc                  C   s>   t ƒ } | dk r	dS zdd l}W n
 ty   Y dS w tj ¡ S )NéZ   Fr   )r   Ú	deep_gemmÚImportErrorr   ÚSGLANG_ENABLE_JIT_DEEPGEMMÚget)Ú
sm_versionr   © r   úb/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/layers/deep_gemm_wrapper/configurer.pyÚ_compute_enable_deep_gemm	   s   ÿ
r   )ÚloggingÚsglang.srt.environr   Úsglang.srt.utilsr   r   Ú	getLoggerÚ__name__Úloggerr   ÚENABLE_JIT_DEEPGEMMÚDEEPGEMM_BLACKWELLÚDEEPGEMM_SCALE_UE8M0r   r   r   r   Ú<module>   s    

