mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-24 01:29:57 +08:00
@@ -36,8 +36,12 @@ import triton.language as tl
|
||||
from paddleformers.utils.log import logger
|
||||
|
||||
from fastdeploy.config import FDConfig, RoutingReplayConfig
|
||||
from fastdeploy.model_executor.ops.triton_ops.triton_utils import (
|
||||
enable_compat_on_triton_kernel,
|
||||
)
|
||||
|
||||
|
||||
@enable_compat_on_triton_kernel
|
||||
@triton.jit
|
||||
def _save_routing_kernel(
|
||||
ROUTING_REPLAY_TABLE_PTR,
|
||||
|
||||
Reference in New Issue
Block a user