update usage of trtllm_fp8_per_tensor_scale_moe (#12569)
This commit is contained in:
@@ -689,7 +689,7 @@ class ModelOptFp8MoEMethod(FusedMoEMethodBase):
|
||||
else 1.0
|
||||
),
|
||||
use_routing_scales_on_input=use_routing_scales_on_input,
|
||||
tile_tokens_dim=8, # TODO(brayden): use the FI tile calculation
|
||||
tile_tokens_dim=None,
|
||||
routing_method_type=routing_method_type,
|
||||
)
|
||||
sm.tag(output)
|
||||
|
||||
Reference in New Issue
Block a user