"vscode:/vscode.git/clone" did not exist on "086b96339ffe057f92cd0a20c8be820b17e24dbf"
Commit 4ad7a1fe authored by lixh6's avatar lixh6
Browse files

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

parent 18459e7a
...@@ -9,7 +9,7 @@ import math ...@@ -9,7 +9,7 @@ import math
from collections.abc import Callable from collections.abc import Callable
from typing import Any from typing import Any, Callable, Dict, List, Optional
import torch import torch
...@@ -1613,6 +1613,8 @@ def fused_experts( ...@@ -1613,6 +1613,8 @@ def fused_experts(
quant_config: FusedMoEQuantConfig | None = None, quant_config: FusedMoEQuantConfig | None = None,
use_nn_moe: bool | None = False, use_nn_moe: bool | None = False,
use_fused_gate: bool | None = False, use_fused_gate: bool | None = False,
i_q: Optional[torch.Tensor] = None,
i_s: Optional[torch.Tensor] = None, **_
) -> torch.Tensor: ) -> torch.Tensor:
if quant_config is None: if quant_config is None:
quant_config = FUSED_MOE_UNQUANTIZED_CONFIG 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