U
    ‰d•  ã                   @   s   d Z ddlmZ dd„ ZdS )z¢
The HIP test utils is a small addition on top of the hypothesis test utils
under caffe2/python, which allows one to more easily test HIP/ROCm related
operators.
é    )Ú
caffe2_pb2c                 C   s    | j tjkptjdd„ |D ƒkS )Nc                 S   s   h | ]
}|j ’qS © )Údevice_type)Ú.0Údr   r   ú?/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/hip_test_util.pyÚ	<setcomp>   s     zrun_in_hip.<locals>.<setcomp>)r   r   ZHIP)ÚgcÚdcr   r   r   Ú
run_in_hip   s    ÿr   N)Ú__doc__Zcaffe2.protor   r   r   r   r   r   Ú<module>   s   