add prints

This commit is contained in:
Michael Melesse
2022-10-24 18:28:28 +00:00
parent 8da4323514
commit 9184b5cf65
3 changed files with 23 additions and 10 deletions

View File

@@ -109,7 +109,7 @@ def check_type_supported(dtype):
pytest.skip("bfloat16 is only supported on NVGPU with cc >= 80")
@pytest.mark.parametrize("dtype_x", [dtype_x for dtype_x in dtypes] + ["bfloat16"])
@pytest.mark.parametrize("dtype_x", [dtype_x for dtype_x in dtypes])
def test_empty_kernel(dtype_x, device='cuda'):
SIZE = 128