Skip to content

Commit 8f1b28c

Browse files
committed
updated name
Signed-off-by: Kinjal Patel <kinjalpravin@nvidia.com>
1 parent 1e11c9b commit 8f1b28c

File tree

1 file changed

+1
-1
lines changed
  • modelopt/torch/quantization/plugins

1 file changed

+1
-1
lines changed

modelopt/torch/quantization/plugins/vllm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,7 @@ def _setup(self):
336336
self.parallel_state = create_parallel_state()
337337

338338
if getattr(self, "invoke_fused_moe_kernel_func", None) is None: # pragma: no cover
339-
for name in ("invoke_fused_moe_kernel", "invoke_fused_moe_triton_kernel"):
339+
for name in ("invoke_fused_moe_kernel", "dispatch_fused_moe_kernel"):
340340
if hasattr(vllm_fused_moe_package, name):
341341
self.invoke_fused_moe_kernel_func = name
342342
break

0 commit comments

Comments
 (0)