Skip to content

Commit df1ae03

Browse files
committed
Minor fix of passing activation_type in test_trtllm_gen_fused_moe.py
Signed-off-by: amitz-nv <203509407+amitz-nv@users.noreply.github.com>
1 parent 67c0ea4 commit df1ae03

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/moe/test_trtllm_gen_fused_moe.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2164,7 +2164,7 @@ def dequant_reference_dsfp8(input, scale, transpose_scale, block_m, block_n):
21642164
gemm2_weights_dequant,
21652165
args.permute_info,
21662166
args.use_routing_scales_on_input,
2167-
args.activation_type.value,
2167+
args.activation_type,
21682168
)
21692169

21702170
return run_moe_dequant(args_dequant, QuantMode.FP8_BLOCK_SCALE), args_dequant
@@ -2201,7 +2201,7 @@ def run_moe_reference_per_tensor_scale_fp8(args):
22012201
gemm2_weights_dequant,
22022202
args.permute_info,
22032203
args.use_routing_scales_on_input,
2204-
args.activation_type.value,
2204+
args.activation_type,
22052205
)
22062206

22072207
return run_moe_dequant(args_dequant, QuantMode.FP8_PER_TENSOR), args_dequant
@@ -2232,7 +2232,7 @@ def run_moe_reference_bf16(args):
22322232
gemm2_weights_dequant,
22332233
args.permute_info,
22342234
args.use_routing_scales_on_input,
2235-
args.activation_type.value,
2235+
args.activation_type,
22362236
)
22372237

22382238
return run_moe_dequant(args_dequant, QuantMode.BF16), args_dequant

0 commit comments

Comments
 (0)