move more files under srt/utils (#11285)
This commit is contained in:
@@ -7,7 +7,6 @@ from pathlib import Path
|
||||
import torch
|
||||
import triton
|
||||
|
||||
from sglang.srt.bench_utils import bench_kineto
|
||||
from sglang.srt.layers.quantization.fp8_kernel import (
|
||||
create_per_token_group_quant_fp8_output_scale,
|
||||
)
|
||||
@@ -16,6 +15,7 @@ from sglang.srt.layers.quantization.fp8_kernel import (
|
||||
)
|
||||
from sglang.srt.layers.quantization.fp8_kernel import sglang_per_token_group_quant_8bit
|
||||
from sglang.srt.utils import is_hip
|
||||
from sglang.srt.utils.bench_utils import bench_kineto
|
||||
|
||||
# CI environment detection
|
||||
IS_CI = (
|
||||
|
||||
@@ -11,7 +11,7 @@ from sgl_kernel.testing.rotary_embedding import (
|
||||
create_inputs,
|
||||
)
|
||||
|
||||
from sglang.srt.bench_utils import bench_kineto
|
||||
from sglang.srt.utils.bench_utils import bench_kineto
|
||||
|
||||
# CI environment detection
|
||||
IS_CI = (
|
||||
|
||||
Reference in New Issue
Block a user