a
    ==icv  ã                   @   sþ   d Z ddlmZ ddlmZ ddlmZ ddlmZ	 e	j
Z
e	jZejejdZe	jejjr`egng  Zdd„ Zd	d
„ Zeeƒ eƒ dZee e
g¡e e
g¡dZee eg¡e eg¡dZee eg¡e eg¡dZee e
eg¡e e
eg¡dZdS )z
The MKL test utils is a small addition on top of the hypothesis test utils
under caffe2/python, which allows one to more easily test MKL related
operators.
é    N)Ú
caffe2_pb2)Ú	workspace)Úhypothesis_test_util)Zdevice_typec                   C   s
   t  t¡S ©N)ÚstÚjustÚdevice_options© r	   r	   úl/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/mkl_test_util.pyÚdevice_checker_device_options   s    r   c                   C   s
   t  t¡S r   )r   Úsampled_fromr   r	   r	   r	   r
   Úgradient_checker_device_option   s    r   )ÚgcZdc)Ú__doc__Zhypothesis.strategiesÚ
strategiesr   Zcaffe2.protor   Zcaffe2.pythonr   r   ÚhuZcpu_doZgpu_doZDeviceOptionZMKLDNNZmkl_dor   ÚCZ
has_mkldnnr   r   ÚdictZgcsr   r   Zgcs_cpu_onlyZgcs_gpu_onlyZgcs_mkl_onlyZgcs_cpu_mklr	   r	   r	   r
   Ú<module>   s(   ÿþ