[TEST] Skip test_dot fp16 out_dtype test (triton-lang#17)

goostavz · Jul 24, 2023 · ef54be0 · ef54be0
1 parent 0838995
commit ef54be0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/TODO.md b/TODO.md
@@ -39,3 +39,4 @@ https://github.com/openai/triton-hopper/blob/1ada046fdaef13f94dc7e2f6e6d0966e5d1
 https://github.com/openai/triton-hopper/blob/b6a6b32b0ee79e93247d20c95f15fd75039a40b9/python/triton/compiler/utils.py#L3
 * Pipeline doesn't handle block ptrs correctly
 * Pipeline doesn't handle TMAs correctly
+* `wgmma` doesn't support `out_dtype=f16`
diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py
@@ -2117,6 +2117,10 @@ def test_dot(M, N, K, num_warps, col_a, col_b, epilogue, allow_tf32, in_dtype, o
         if out_dtype == 'float16':
             # TODO: support out_dtype=float16 for tl.dot on V100
             pytest.skip("Only test out_dtype=float16 on devices with sm >=80")
+    if capability[0] == 9 and out_dtype == 'float16':
+        # TODO: support out_dtype=float16 for tl.dot on H100
+        pytest.skip("Only test out_dtype=float16 on devices with sm<90")
+
 
     torch.backends.cuda.matmul.allow_tf32 = allow_tf32