Jia Guo ec01ef9092 Fix torch.compile/dynamo crash with Qwen3 QK-norm in piecewise CUDA g… (#19818) ...

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

2026-03-20 18:05:09 +08:00

..

benchmark

Replace _resolve_future_token_ids with JIT kernel + platform dispatch (#20976)

2026-03-20 01:47:03 -07:00

csrc

Replace _resolve_future_token_ids with JIT kernel + platform dispatch (#20976)

2026-03-20 01:47:03 -07:00

diffusion

[Diffusion] Opt qwen-image-edit with fuse_residual_layernorm_scale_shift_gate_select01_kernel (#20395)

2026-03-13 13:15:22 +08:00

include/sgl_kernel

[JIT] Inject target architecture flag into JIT compilation (#20103)

2026-03-17 23:16:49 -07:00

tests

Replace _resolve_future_token_ids with JIT kernel + platform dispatch (#20976)

2026-03-20 01:47:03 -07:00

__init__.py

…

__main__.py

…

.clang-format

…

add_constant.py

[1/n jit_kernel restruct] unify cache usage and clean up naming in ngram_embedding (#20244)

2026-03-10 15:53:43 +08:00

awq_dequantize.py

[Kernel Slimming] Migrate AWQ marlin repack kernel to JIT (#18949)

2026-02-23 22:05:27 +08:00

awq_marlin_repack.py

[Kernel Slimming] Migrate AWQ marlin repack kernel to JIT (#18949)

2026-02-23 22:05:27 +08:00

concat_mla.py

…

cutedsl_gdn.py

…

flash_attention_v4.py

bump fa4 to official released fa4 pkg (#20303)

2026-03-17 17:22:56 -07:00

fused_metadata_copy.py

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

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

fused_store_index_cache.py

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

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

gptq_marlin_repack.py

[Kernel Slimming] Migrate GPTQ-Marlin repack kernel to JIT (#18543)

2026-02-13 22:29:22 +08:00

gptq_marlin.py

…

hadamard.py

refactor(jit_kernel): reduce duplication and separate test code (#19323)

2026-02-26 18:30:49 +08:00

hicache.py

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

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

kvcache.py

…

moe_lora_align.py

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

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

moe_wna16_marlin.py

[Kernel Slimming] Migrate marlin moe kernel to JIT (#19181)

2026-02-26 09:05:13 +08:00

ngram_embedding.py

[1/n jit_kernel restruct] unify cache usage and clean up naming in ngram_embedding (#20244)

2026-03-10 15:53:43 +08:00

norm.py

Fix torch.compile/dynamo crash with Qwen3 QK-norm in piecewise CUDA g… (#19818)

2026-03-20 18:05:09 +08:00

nvfp4.py

[JIT Kernel] Fix NVFP4 multi-arch compilation failure (#20874)

2026-03-20 10:30:04 +08:00

per_tensor_quant_fp8.py

…

per_token_group_quant_8bit.py

[Re-land][jit kernel] Support per_token_group_quant_8bit jit kernel (#19140)

2026-02-26 09:53:57 +08:00

resolve_future_token_ids.py

Replace _resolve_future_token_ids with JIT kernel + platform dispatch (#20976)

2026-03-20 01:47:03 -07:00

rope.py

[2/n jit_kernel restruct] unify rotary embedding entrypoints under rope.py (#20247)

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

timestep_embedding.py

…

utils.py

[JIT] Inject target architecture flag into JIT compilation (#20103)

2026-03-17 23:16:49 -07:00