mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
ccfaf46431
Summary: This is mostly for reusing all the cudnn test cases in our python operator_tests. Pull Request resolved: https://github.com/pytorch/pytorch/pull/12278 Differential Revision: D10842592 Pulled By: bddppq fbshipit-source-id: 4b3ed91fca64ff02060837b3270393bc2f9a9898
19 lines
550 B
Python
19 lines
550 B
Python
## @package hip_test_util
|
|
# Module caffe2.python.hip_test_util
|
|
"""
|
|
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.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
from caffe2.proto import caffe2_pb2
|
|
|
|
def run_in_hip(gc, dc):
|
|
return (gc.device_type == caffe2_pb2.HIP) or (
|
|
caffe2_pb2.HIP in {d.device_type for d in dc})
|