[FRONTEND] Fix scanline layout (#548)

2022-06-13 16:21:10 -07:00
parent 7094657aa9
commit 58c8889235
2 changed files with 7 additions and 10 deletions
--- a/python/test/unit/language/test_core.py
+++ b/python/test/unit/language/test_core.py
@@ -691,7 +691,7 @@ def test_f16_to_f8_rounding():
@pytest.mark.parametrize("dtype_str, shape",
                         [(dtype, shape)
                          for dtype in dtypes
-                          for shape in [128, 512]])
+                          for shape in [32, 64, 128, 512]])
 def test_reduce1d(dtype_str, shape, device='cuda'):

    # triton kernel