We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 434609a commit 7cc74c4Copy full SHA for 7cc74c4
tensorrt_llm/_torch/custom_ops/trtllm_gen_custom_ops.py
@@ -449,8 +449,13 @@ def fp4_block_scale_moe_runner(
449
hidden_states_scale,
450
gemm1_weights,
451
gemm1_weights_scale,
452
+ gemm1_bias,
453
+ gemm1_alpha,
454
+ gemm1_beta,
455
+ gemm1_clamp_limit,
456
gemm2_weights,
457
gemm2_weights_scale,
458
+ gemm2_bias,
459
output1_scale_scalar,
460
output1_scale_gate_scalar,
461
output2_scale_scalar,
0 commit comments