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

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

..

diffusion

[Diffusion] Add diffusion time embedding to jit kernel (#17658)

2026-01-24 14:27:08 +08:00

distributed

[JIT Kernel][Feature] Support JIT custom all reduce (rewrite as v2) (#19880)

2026-03-20 18:24:07 +08:00

elementwise

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

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

fast-hadamard-transform

Remove unused fast-hadamard-transform PyTorch extension sources (#18927)

2026-02-18 15:51:07 +08:00

gemm

[JIT Kernel][Feature] Support JIT custom all reduce (rewrite as v2) (#19880)

2026-03-20 18:24:07 +08:00

lora

[LoRA][I] Add MOE LoRA JIT alignment kernel and tests (#19710)

2026-03-12 12:23:46 -07:00

moe

[JIT Kernel] Reland NVFP4 kernels to JIT (#20012)

2026-03-07 10:31:08 +08:00

nsa

[DeepSeek-V3.2][JIT-kernel] Support nsa fuse store indexer k cache (#19148)

2026-02-26 10:23:10 +08:00

add_constant.cuh

[Refactor] Clean up JIT kernel utilites (#16884)

2026-01-13 17:54:16 +08:00

hicache.cuh

[Refactor] Fix test and clean up hicache code (#18555)

2026-02-18 14:37:46 +08:00

hisparse.cuh

HiSparse for Sparse Attention (#20343)

2026-03-22 23:09:31 -07:00

ngram_embedding.cuh

Feature/support longcat flash lite (#17838)

2026-03-09 23:00:11 +08:00