Skip to content

Commit e15a96c

Browse files
committed
update to address comment
1 parent 91a85ad commit e15a96c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

csrc/nv_internal/tensorrt_llm/kernels/cutlass_kernels/include/moe_util_kernels.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,8 @@ void expandInputRowsKernelLauncher(
6464
int const k, int const num_experts_per_node, QuantParams const& quant_params,
6565
bool use_per_expert_act_scale, int64_t* expert_first_token_offset,
6666
TmaWarpSpecializedGroupedGemmInput::ElementSF* fc1_act_sf_flat,
67-
TmaWarpSpecializedGroupedGemmInput::ElementSF const* input_sf, void const* prequant_scales,
68-
bool enable_pdl, cudaStream_t stream);
67+
TmaWarpSpecializedGroupedGemmInput::ElementSF const* input_sf, bool const swizzled_input_sf,
68+
void const* prequant_scales, bool enable_pdl, cudaStream_t stream);
6969

7070
template <class OutputType, class GemmOutputType, class ScaleBiasType>
7171
void finalizeMoeRoutingKernelLauncher(

0 commit comments

Comments
 (0)