[FRONTEND] Fix scanline layout (#548)

This commit is contained in:
Philippe Tillet
2022-06-13 16:21:10 -07:00
committed by GitHub
parent 7094657aa9
commit 58c8889235
2 changed files with 7 additions and 10 deletions

View File

@@ -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