| 123456789101112131415161718 |
- ## @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 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})
|
