Skip to content

Commit c187427

Browse files
committed
remove comment
Signed-off-by: jiahanc <[email protected]>
1 parent f5f1c55 commit c187427

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

csrc/trtllm_batched_gemm_runner.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ void TrtllmGenBatchedGemmRunner::run(
169169
auto const configs = bmm.getBatchedGemmConfigs();
170170

171171
auto const& config = configs[configIndex];
172-
// std::cout << "Running GEMM with config: " << config.mFunctionName << std::endl;
172+
173173
FLASHINFER_CHECK(numBatches > 0, "Batched GEMM requires numBatches > 0");
174174
if (!mOptions.staticBatch) {
175175
FLASHINFER_CHECK(totalNumPaddedTokens,

flashinfer/fused_moe/core.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1345,7 +1345,6 @@ def trtllm_fp4_block_scale_moe_op(
13451345
if hidden_states_scale is not None:
13461346
inputs.append(hidden_states_scale)
13471347

1348-
print(f"fp4 block scale moe tunning start")
13491348
_, tactic = tuner.choose_one(
13501349
"flashinfer::trtllm_fp4_block_scale_moe",
13511350
[moe_runner],
@@ -1374,7 +1373,7 @@ def trtllm_fp4_block_scale_moe_op(
13741373
do_finalize=do_finalize,
13751374
gated_act_type=gated_act_type,
13761375
)
1377-
print(f"fp4 block scale moe tunning end with tactic {tactic}")
1376+
13781377
# Call the C++ function for block scale MoE
13791378
output = moe_op.trtllm_fp4_block_scale_moe(
13801379
routing_logits,

0 commit comments

Comments
 (0)