Skip to content

Commit 7cc74c4

Browse files
committed
Fix rebase issue
Signed-off-by: Dongfeng Yu <[email protected]>
1 parent 434609a commit 7cc74c4

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tensorrt_llm/_torch/custom_ops/trtllm_gen_custom_ops.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -449,8 +449,13 @@ def fp4_block_scale_moe_runner(
449449
hidden_states_scale,
450450
gemm1_weights,
451451
gemm1_weights_scale,
452+
gemm1_bias,
453+
gemm1_alpha,
454+
gemm1_beta,
455+
gemm1_clamp_limit,
452456
gemm2_weights,
453457
gemm2_weights_scale,
458+
gemm2_bias,
454459
output1_scale_scalar,
455460
output1_scale_gate_scalar,
456461
output2_scale_scalar,

0 commit comments

Comments
 (0)