Fix sgl-kernel benchmark dead code (#11022)
This commit is contained in:
@@ -1,9 +1,22 @@
|
||||
import os
|
||||
|
||||
import torch
|
||||
|
||||
# CI environment detection
|
||||
IS_CI = (
|
||||
os.getenv("CI", "false").lower() == "true"
|
||||
or os.getenv("GITHUB_ACTIONS", "false").lower() == "true"
|
||||
)
|
||||
import triton
|
||||
|
||||
from sglang.srt.layers.moe.ep_moe.kernels import post_reorder_triton_kernel
|
||||
|
||||
batch_sizes = [64, 128, 256, 512, 640, 768, 1024, 2048, 4096]
|
||||
# CI environment uses simplified parameters
|
||||
if IS_CI:
|
||||
batch_sizes = [64, 128] # Only test 2 values in CI
|
||||
else:
|
||||
batch_sizes = [64, 128, 256, 512, 640, 768, 1024, 2048, 4096]
|
||||
|
||||
configs = [(bs,) for bs in batch_sizes]
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user