diff --git a/tests/cpp/test_combined_inner_outer_reduction.cpp b/tests/cpp/test_combined_inner_outer_reduction.cpp index e862cf74d6d..9115c7ed8e8 100644 --- a/tests/cpp/test_combined_inner_outer_reduction.cpp +++ b/tests/cpp/test_combined_inner_outer_reduction.cpp @@ -1174,7 +1174,13 @@ TEST_P(TmaWarpSpecializedTest, RMSNormBwd) { TEST_P(TmaWarpSpecializedTest, ThunderRMSNormBwd) { NVFUSER_TEST_CUDA_ARCH_GUARD(9, 0); - auto [contig, _, dtype, dim0, dim1] = GetParam(); + auto [contig, ws_enabled, dtype, dim0, dim1] = GetParam(); + + if (ws_enabled) { + GTEST_SKIP() << "Bdimx is dynamic, Warp Specialization is disabled."; + return; + } + std::vector norm_shape{dim1}; auto fusion = std::make_unique(); @@ -1231,7 +1237,12 @@ TEST_P(TmaWarpSpecializedTest, LayerNormBackward) { NVFUSER_TEST_CUDA_ARCH_GUARD(9, 0); auto fusion = std::make_unique(); FusionGuard fg(fusion.get()); - auto [contig, _, dtype, dim0, dim1] = GetParam(); + auto [contig, ws_enabled, dtype, dim0, dim1] = GetParam(); + + if (ws_enabled) { + GTEST_SKIP() << "Bdimx is dynamic, Warp Specialization is disabled."; + return; + } std::vector norm_shape{dim1}; std::vector input_shape{dim0, dim1};