fix minor bug

2022-10-31 18:33:47 +00:00
parent 14730a2352
commit 9b3f2487b5
1 changed files with 6 additions and 3 deletions
--- a/python/test/unit/language/test_core.py
+++ b/python/test/unit/language/test_core.py
@@ -104,6 +104,9 @@ def check_type_supported(dtype):
    '''
    skip test if dtype is not supported on the current device
    '''
+    if torch.version.hip is not None:
+        pass
+    else:
        cc = _triton.runtime.cc(_triton.runtime.backend.CUDA, torch.cuda.current_device())
        if cc < 80 and (dtype is tl.bfloat16 or dtype == "bfloat16" or dtype is torch.bfloat16):
            pytest.skip("bfloat16 is only supported on NVGPU with cc >= 80")