diff --git a/test/70.program/test_program_with_micro_kernel.py b/test/70.program/test_program_with_micro_kernel.py index a95017f92..b0825e793 100644 --- a/test/70.program/test_program_with_micro_kernel.py +++ b/test/70.program/test_program_with_micro_kernel.py @@ -5,7 +5,7 @@ @pytest.mark.skipif(not ft.with_pytorch() or not ft.with_cuda(), reason="requires PyTorch and CUDA") -def test_matmul(): +def test_matmul_float64(): M = N = K = 5000 block_n = block_m = 128