File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -169,7 +169,7 @@ void TrtllmGenBatchedGemmRunner::run(
169
169
auto const configs = bmm.getBatchedGemmConfigs ();
170
170
171
171
auto const & config = configs[configIndex];
172
- // std::cout << "Running GEMM with config: " << config.mFunctionName << std::endl;
172
+
173
173
FLASHINFER_CHECK (numBatches > 0 , " Batched GEMM requires numBatches > 0" );
174
174
if (!mOptions .staticBatch ) {
175
175
FLASHINFER_CHECK (totalNumPaddedTokens,
Original file line number Diff line number Diff line change @@ -1345,7 +1345,6 @@ def trtllm_fp4_block_scale_moe_op(
1345
1345
if hidden_states_scale is not None :
1346
1346
inputs .append (hidden_states_scale )
1347
1347
1348
- print (f"fp4 block scale moe tunning start" )
1349
1348
_ , tactic = tuner .choose_one (
1350
1349
"flashinfer::trtllm_fp4_block_scale_moe" ,
1351
1350
[moe_runner ],
@@ -1374,7 +1373,7 @@ def trtllm_fp4_block_scale_moe_op(
1374
1373
do_finalize = do_finalize ,
1375
1374
gated_act_type = gated_act_type ,
1376
1375
)
1377
- print ( f"fp4 block scale moe tunning end with tactic { tactic } " )
1376
+
1378
1377
# Call the C++ function for block scale MoE
1379
1378
output = moe_op .trtllm_fp4_block_scale_moe (
1380
1379
routing_logits ,
You can’t perform that action at this time.
0 commit comments