Skip to content

Commit da54367

Browse files
committed
debug..
1 parent bfe2852 commit da54367

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/moe_gemm/moe_gemm_template_dispatch_tma_ws.h

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,14 @@ void dispatchMoeGemmFinalDispatchTmaWarpSpecialized(
189189
auto cluster_shape_cute_fallback = cute::Shape<int32_t, int32_t, cute::_1>{
190190
std::get<0>(cluster_shape_fallback), std::get<1>(cluster_shape_fallback), cute::_1{}};
191191

192+
// HACK debug the gemm_config used to produce selected_func
193+
std::cout << "[SM100 gemm_config] sm_version=" << gemm_config.sm_version
194+
<< ", tile_config_sm100=" << static_cast<int>(gemm_config.tile_config_sm100)
195+
<< ", epilogue_schedule=" << static_cast<int>(gemm_config.epilogue_schedule)
196+
<< ", dynamic_cluster_shape=" << static_cast<int>(gemm_config.dynamic_cluster_shape)
197+
<< ", fallback_cluster_shape="
198+
<< static_cast<int>(gemm_config.fallback_cluster_shape) << std::endl;
199+
192200
auto selected_func =
193201
getDispatchFunctionForSM100<Arch, T, WeightType, OutputType, EpilogueTag, FUSION,
194202
TileShape, ClusterShape, is_wfp4afp8>(

0 commit comments

Comments
 (0)