o
    HŠ!iÂ  ã                   @   s0   d dl mZmZ d dlmZ dd„ Zdd„ ZdS )é   )Úget_default_deviceÚparse_device)Úget_device_typec                  C   s   ddl m}  t| dƒS )Né    ©ÚConfigZset_mkldnn_cache_capacity)Zpaddle.inferencer   Úhasattrr   © r	   úh/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddlex/inference/utils/misc.pyÚis_mkldnn_available   s   
r   c                 C   s>   dd l }| d u rtƒ } t| ƒ\}}dtƒ v p|j ¡ o|dv S )Nr   Únpu)Zgpur   ZxpuZmlu)Z
paddle.ampr   r   r   ÚampZis_bfloat16_supported)ZdeviceZpaddleZdevice_typeÚ_r	   r	   r
   Úis_bfloat16_available   s   ÿþr   N)Zutils.devicer   r   Z	utils.envr   r   r   r	   r	   r	   r
   Ú<module>   s   