diff --git a/cpp/tensorrt_llm/kernels/cutlass_kernels/fp4_gemm/fp4_gemm_template.h b/cpp/tensorrt_llm/kernels/cutlass_kernels/fp4_gemm/fp4_gemm_template.h index dd7e8c6bb9..de59081bd0 100644 --- a/cpp/tensorrt_llm/kernels/cutlass_kernels/fp4_gemm/fp4_gemm_template.h +++ b/cpp/tensorrt_llm/kernels/cutlass_kernels/fp4_gemm/fp4_gemm_template.h @@ -440,7 +440,7 @@ std::vector CutlassFp4GemmRunner::getCon } } CutlassGemmConfig config(tile_config, tkc::MainloopScheduleType::AUTO, tkc::EpilogueScheduleType::AUTO, - cluster_config, ClusterShape::Undefined, ClusterShape::Undefined, mSm); + cluster_config, tkc::ClusterShape::Undefined, tkc::ClusterShape::Undefined, mSm); candidateConfigs.push_back(config); } }