[TESTS] Added bfloat16 tests (#430)

2022-01-14 15:38:32 +08:00
parent 4c94359199
commit 2a944ded53
6 changed files with 22 additions and 6 deletions
--- a/python/test/unit/language/test_core.py
+++ b/python/test/unit/language/test_core.py
@@ -732,6 +732,8 @@ def test_dot(epilogue, allow_tf32, device='cuda'):
    assert 'st.global.v4' in ptx
    if allow_tf32:
        assert 'mma.sync.aligned.m16n8k8.row.col.f32.tf32.tf32.f32' in ptx
+    else:
+        assert 'mma.sync.aligned.m16n8k8.row.col.f32.tf32.tf32.f32' not in ptx


 def test_dot_without_load():