From 49b086249f10f5d3cd15675e5abaaae39a0e505a Mon Sep 17 00:00:00 2001 From: Jingyue Wu Date: Thu, 19 Dec 2024 20:54:13 -0800 Subject: [PATCH] More matmul test coverage (#3622) --- tests/cpp/test_matmul_scheduler.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/cpp/test_matmul_scheduler.cpp b/tests/cpp/test_matmul_scheduler.cpp index 540015e709a..c860b908a92 100644 --- a/tests/cpp/test_matmul_scheduler.cpp +++ b/tests/cpp/test_matmul_scheduler.cpp @@ -1060,7 +1060,7 @@ TEST_F(MatmulSchedulerTest, FusedMultiplySumOnly) { // for Ampere with strict ref check, hence single layout check TEST_F(MatmulSchedulerTest, BasicMatmulStrictCheckTT) { // TODO: Make these tests work with Hopper as well as Ampere - NVFUSER_TEST_CUDA_ARCH_RANGE_GUARD(8, 0, 8, 9); + NVFUSER_TEST_CUDA_ARCH_RANGE_GUARD(8, 0, 9, 0); const int M = 128, N = 256, K = 512; const auto layout = MmaLayout::TT; @@ -2481,7 +2481,7 @@ class MatmulSchedulerPluginTest : public NVFuserTest { // Test that our fake plugin works to override the default heuristic TEST_F(MatmulSchedulerPluginTest, BasicMatmul) { - NVFUSER_TEST_CUDA_ARCH_RANGE_GUARD(8, 0, 8, 9); + NVFUSER_TEST_CUDA_ARCH_RANGE_GUARD(8, 0, 9, 0); const int M = 128, N = 256, K = 512; const auto layout = MmaLayout::TT; auto fusion = std::make_unique(); @@ -3156,7 +3156,7 @@ INSTANTIATE_TEST_SUITE_P( #undef NVFUSER_TEST_CUDA_ARCH_GUARD TEST_F(MatmulSchedulerTest, OperandOrderIssue2434) { - NVFUSER_TEST_CUDA_ARCH_RANGE_GUARD(8, 0, 8, 9); + NVFUSER_TEST_CUDA_ARCH_RANGE_GUARD(8, 0, 9, 0); int M = 32, N = 64, K = 128; std::unique_ptr fusion_ptr = std::make_unique();