[Kernel] Sweep missed dedicated kernels into kernels.ops (moe/quant siblings + dspark) (RFC #29630) (#32160)
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.8
parent
f35411ee81
commit
62aa85d9aa
@@ -16,16 +16,16 @@ from sglang.kernels.ops.attention import (
|
||||
dsv4_attn_metadata_kernels as attn_metadata_kernels,
|
||||
)
|
||||
from sglang.kernels.ops.speculative import ragged_verify_kernels
|
||||
from sglang.srt.speculative.dspark_components.dspark_planner import (
|
||||
DSparkScheduleConfig,
|
||||
)
|
||||
from sglang.srt.speculative.dspark_components.kernels import (
|
||||
from sglang.kernels.ops.speculative.dspark import (
|
||||
dspark_accept,
|
||||
dspark_attn_metadata,
|
||||
dspark_draft_model,
|
||||
dspark_schedule,
|
||||
dspark_verify_window,
|
||||
)
|
||||
from sglang.srt.speculative.dspark_components.dspark_planner import (
|
||||
DSparkScheduleConfig,
|
||||
)
|
||||
from sglang.srt.speculative.ragged_verify import RaggedVerifyLayout
|
||||
from sglang.test.ci.ci_register import register_cuda_ci
|
||||
from sglang.test.test_utils import CustomTestCase
|
||||
|
||||
Reference in New Issue
Block a user