o
     ¢iÝ  ã                	   @   s  d dl Z d dlmZ z=d dlZd dlZd dlZdZejZeejƒZ	z
ej
j ¡  dZW n ej
jjy9   dZY nw e	jdkrCejZnejZW n eefyb   dZdZdZedƒZ	dZdZdZY nw z>d dlZd dlZdZej
 ¡ d kZeejdƒo€ejj ¡ Zeoˆejj ¡ Z eZ!ee"ejƒƒZ#e#edƒkoŸej
j$j% &¡  Z'W n ey¼   dZdZdZdZ!dZdZ dZ'edƒZ#Y nw d	d
„ Z(da)da*da+dd„ Z,da-da.zd dl/Z/W n eyà   dZ/Y nw zd dl0Z0dZ1W n eyö   dZ0dZ1Y nw epúe Z2g d¢Z3dS )é    N)ÚVersionTFé
   z0.0.0Úmpsz1.9.0c                  C   s<   d} t  | t¡ dd ladd ladattj d¡ƒdka	d S )NzÍBuilt-in TensorFlow support will be removed in Thinc v9. If you need TensorFlow support in the future, you can transition to using a custom copy of the current TensorFlowWrapper in your package or project.r   TÚGPU)
ÚwarningsÚwarnÚDeprecationWarningÚ
tensorflowÚtensorflow.experimental.dlpackÚhas_tensorflowÚlenÚconfigÚget_visible_devicesÚhas_tensorflow_gpu©Úwarn_msg© r   ú@/home/ubuntu/.local/lib/python3.10/site-packages/thinc/compat.pyÚenable_tensorflow;   s   ÿr   c                  C   s    d} t  | t¡ dd ladad S )Nz¾Built-in MXNet support will be removed in Thinc v9. If you need MXNet support in the future, you can transition to using a custom copy of the current MXNetWrapper in your package or project.r   T)r   r   r   ÚmxnetÚ	has_mxnetr   r   r   r   Úenable_mxnetP   s
   ÿr   )ÚcupyÚcupyxÚtorchr	   r   Úh5pyÚos_signpost)4r   Úpackaging.versionr   r   Úcupy.cublasr   Úhas_cupyÚcublasÚ__version__Úcupy_versionÚcudaÚruntimeÚgetDeviceCountÚhas_cupy_gpuÚCUDARuntimeErrorÚmajorÚfrom_dlpackÚcupy_from_dlpackÚ
fromDlpackÚImportErrorÚAttributeErrorr   Útorch.utils.dlpackÚ	has_torchÚdevice_countÚhas_torch_cuda_gpuÚhasattrÚbackendsr   Úis_builtÚhas_torch_mpsÚis_availableÚhas_torch_mps_gpuÚhas_torch_gpuÚstrÚtorch_versionÚampÚcommonÚamp_definitely_not_availableÚhas_torch_ampr   r	   r   r   r   r   r   r   r   Úhas_os_signpostÚhas_gpuÚ__all__r   r   r   r   Ú<module>   sŽ    
ÿ
€ù
þøÿþ