Commit 4ad7a1fe authored by lixh6's avatar lixh6
Browse files

Fix:修复调用Triton MoE gemm时缺失的参数,对齐接口

parent 18459e7a
......@@ -9,7 +9,7 @@ import math
from collections.abc import Callable
from typing import Any
from typing import Any, Callable, Dict, List, Optional
import torch
......@@ -1613,6 +1613,8 @@ def fused_experts(
quant_config: FusedMoEQuantConfig | None = None,
use_nn_moe: bool | None = False,
use_fused_gate: bool | None = False,
i_q: Optional[torch.Tensor] = None,
i_s: Optional[torch.Tensor] = None, **_
) -> torch.Tensor:
if quant_config is None:
quant_config = FUSED_MOE_UNQUANTIZED_CONFIG
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment