[refactor] Move trtllm_fp8_kv_kernel to triton_ops directory (#15044)

2025-12-14 15:22:37 -08:00
parent 0612175ce1
commit 99cb2ed988
3 changed files with 6 additions and 2 deletions
@@ -6,7 +6,9 @@ import unittest

 import torch

-from sglang.srt.layers.attention.trtllm_fp8_kv_kernel import fused_fp8_set_kv_buffer
+from sglang.srt.layers.attention.triton_ops.trtllm_fp8_kv_kernel import (
+    fused_fp8_set_kv_buffer,
+)
 from sglang.test.test_utils import CustomTestCase