[Kernel] Add JIT rotary_embedding_kernel#17934

Merged

BBuf merged 12 commits into

sgl-project:mainfrom

pansicheng:jit-rotary-embedding

Feb 13, 2026

Commits on Jan 29, 2026

Commits on Jan 30, 2026

Change import of rotary_embedding from sgl_kernel to sglang.jit_kernel.pos_enc in SRT with logging and test updates.
pansicheng
committed
Make key tensor optional in rotary embedding kernel
pansicheng
committed
Add PyTorch-native rotary embedding implementation for testing
pansicheng
committed

Commits on Feb 2, 2026

Apply suggestions from code review

pansicheng
and
BBuf
authored
Fix cr
pansicheng
committed

Commits on Feb 3, 2026

Add stream support
pansicheng
committed
Revert "Add stream support"

root
authored and
root
committed

Commits on Feb 4, 2026

Add stream support
pansicheng
committed

Commits on Feb 5, 2026

Register torch custom_op for piecewise cuda graph
pansicheng
committed