[Kernel] Relocate vendored fla and mamba kernel trees to sglang.kernels (RFC #29630, Phase 2.5, 7/7) (#30795)
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Fable 5
parent
23f2b77d82
commit
4aadf94146
@@ -29,10 +29,10 @@ from flashinfer.gdn_prefill import (
|
||||
chunk_gated_delta_rule as flashinfer_chunk_gated_delta_rule,
|
||||
)
|
||||
|
||||
from sglang.srt.layers.attention.fla.chunk import (
|
||||
from sglang.kernels.ops.attention.fla.chunk import (
|
||||
chunk_gated_delta_rule as triton_chunk_gated_delta_rule,
|
||||
)
|
||||
from sglang.srt.layers.attention.fla.l2norm import l2norm_fwd
|
||||
from sglang.kernels.ops.attention.fla.l2norm import l2norm_fwd
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Helpers
|
||||
|
||||
Reference in New Issue
Block a user