[RL] add stream guard (#6814)

* add stream guard

* format
This commit is contained in:
liufengwei0103
2026-03-13 11:22:26 +08:00
committed by GitHub
parent 586e6f38b1
commit 62110045f3
@@ -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,