[FRONTEND] Better dot error message (#531)

2022-05-26 17:41:09 -07:00
parent 0e2883020a
commit c82a206684
3 changed files with 25 additions and 1 deletions
--- a/python/test/unit/language/test_core.py
+++ b/python/test/unit/language/test_core.py
@@ -910,7 +910,7 @@ def test_arange(start, device='cuda'):
 def test_masked_load_shared_memory(dtype, device='cuda'):
    M = 32
    N = 32
-    K = 8
+    K = 16

    in1 = torch.rand((M, K), dtype=dtype, device=device)
    in2 = torch.rand((K, N), dtype=dtype, device=device)