diff --git a/tests/integration/defs/accuracy/test_llm_api_pytorch.py b/tests/integration/defs/accuracy/test_llm_api_pytorch.py index 34688c88513e..bb5222a1d514 100644 --- a/tests/integration/defs/accuracy/test_llm_api_pytorch.py +++ b/tests/integration/defs/accuracy/test_llm_api_pytorch.py @@ -1052,7 +1052,9 @@ def test_cute_dsl_nvfp4( """Test NVFP4 with CuTe DSL blockscaling mm (GEMM+SwiGLU fusion for shared experts).""" sm_version = get_sm_version() if sm_version not in (100, 103): - pytest.skip("CuTe DSL blockscaling mm supports SM 100 and 103 only") + pytest.skip( + "CuTe DSL NVFP4 GEMM+SwiGLU act fusion supports SM 100 and 103 only" + ) kv_cache_config = KvCacheConfig(free_gpu_memory_fraction=0.9) torch_compile_config = _get_default_torch_compile_config(torch_compile) @@ -1104,7 +1106,9 @@ def test_cute_dsl_nvfp4_4gpus( """Test NVFP4 4 GPUs with CuTe DSL blockscaling mm (GEMM+SwiGLU fusion for shared experts).""" sm_version = get_sm_version() if sm_version not in (100, 103): - pytest.skip("CuTe DSL blockscaling mm supports SM 100 and 103 only") + pytest.skip( + "CuTe DSL NVFP4 GEMM+SwiGLU act fusion supports SM 100 and 103 only" + ) kv_cache_config = KvCacheConfig(free_gpu_memory_fraction=0.9) torch_compile_config = _get_default_torch_compile_config(torch_compile)