Johnsonms 777edb6ef7 Fix(jit): support rmsnorm for hidden_size in {64, 128, 256} (#20661)

2026-03-23 23:17:44 +08:00

..

clamp_position.cuh

Enable JIT clamp_position and resolve_future_token_ids on ROCm (#21116)

2026-03-22 22:33:54 -07:00

concat_mla.cuh

[Move sgl-kernel Kernel to JIT] Add JIT concat MLA kernels (#17889)

2026-02-03 10:49:17 +08:00

fused_add_rmsnorm.cuh

Add compile-time 256-bit vector guard for pre-Blackwell (#19794)

2026-03-20 18:25:12 +08:00

fused_metadata_copy.cuh

Kernel: optimize decoding metadata in NSA multi-spec backend with fused kernels (#17554)

2026-02-14 16:40:15 +08:00

kvcache.cuh

[AMD] Enable ROCm kvcache JIT path and add AMD CI coverage. (#18992)

2026-02-25 14:15:05 +08:00

pos_enc.cuh

[Kernel] Add JIT rotary_embedding_kernel (#17934)

2026-02-13 12:41:25 +08:00

qknorm_across_heads.cuh

Add compile-time 256-bit vector guard for pre-Blackwell (#19794)

2026-03-20 18:25:12 +08:00

qknorm.cuh

[JIT kernel] hd=512,1024 in JIT QK norm (cta based) (#17515)

2026-02-16 16:07:24 +08:00

resolve_future_token_ids.cuh

Enable JIT clamp_position and resolve_future_token_ids on ROCm (#21116)

2026-03-22 22:33:54 -07:00

rmsnorm.cuh

Fix(jit): support rmsnorm for hidden_size in {64, 128, 256} (#20661)

2026-03-23 23:17:44 +08:00

rope.cuh

fix(rope): restore K writeback in fused rope + kv store kernel (#19636)

2026-03-07 20:41:35 +08:00