pansicheng 7d4ae057ec [Kernel] Add JIT rotary_embedding_kernel (#17934) ...

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
Co-authored-by: root <root@zhikuan-A10x2.ea134>

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

..

attention

[AMD] Fix accuracy issue when running TP4 dsv3 model with mtp (#18607)

2026-02-12 01:13:16 -08:00

deep_gemm_wrapper

[DeepGemm] Add a flag for fast warmup (#18111)

2026-02-04 14:12:13 +08:00

moe

[Flashinfer Autotune] Fix FlashInfer FP4 MoE autotuning crash by removing incorrect flatten on hidden_states_scale (#18500)

2026-02-12 13:31:27 +08:00

quantization

[Ascend]Support qwen3.5 (#18544)

2026-02-12 15:22:47 +08:00

utils

[MUSA][4/N] Add common device utilities, distributed backend, and custom op wiring (#17246)

2026-01-28 23:13:24 -08:00

activation.py

[NPU] support model skywork-reward-gemma2-2-27B-v0.2 (#16947)

2026-02-11 15:34:53 +08:00

amx_utils.py

[CPU][INT4] Add INT4 kernels for CPU (#8226)

2026-01-29 22:30:13 -08:00

communicator_nsa_cp.py

[DeepSeek v3.2] opt Context Parallelism: support fused moe, multi batch and fp8 kvcache (#13959)

2026-01-02 23:49:14 +08:00

communicator.py

[hotfix] Reenable all reduce fusion on sm100 (#17591)

2026-01-22 23:36:38 +08:00

dp_attention.py

Make sure we always disable symm memory without dp padding (#18129)

2026-02-04 19:58:28 +08:00

elementwise.py

[Auto Sync] Update elementwise.py (20260131) (#18033)

2026-02-01 07:39:37 -08:00

flashinfer_comm_fusion.py

[Refactor] Clean up custom op (#15995)

2026-01-01 21:41:56 +08:00

int4fp8_utils.py

[AMD][Quantization] Add int4fp8_moe online quantization on ROCm (#7392)

2026-01-14 01:44:40 -08:00

layernorm.py

[AMD] rocm 7.2 image release, PR test, Nightly Test (#17799)

2026-02-11 21:29:25 -08:00

linear.py

fuse qkvbfg linear into one gemm and f_b g_b into batched gemm. (#17801)

2026-02-04 11:41:26 +08:00

logits_processor.py

[Auto Sync] Update weight_utils.py (20260212) (#18692)

2026-02-12 16:26:05 -08:00

model_parallel.py

[Refactor] Rename CustomOp -> MultiPlatformOp (#16175)

2025-12-31 01:16:32 +08:00

modelopt_utils.py

Enable native ModelOpt quantization support (1/3) (#7149)

2025-10-06 13:24:15 -07:00

multimodal.py

[VLM] Improving multimodal tensor hash kernel (#9008)

2025-08-14 13:45:55 -07:00

parameter.py

Code clean up for fp8 quantization (#16982)

2026-01-13 12:38:39 -08:00

pooler.py

Support piecewise CUDA graph for embedding models (#13852)

2025-11-26 15:29:46 +08:00

radix_attention.py

feature: support bidirectional attention for Gemma-3 (#10707)

2026-02-09 23:17:45 +08:00

radix_linear_attention.py

[hybrid-model] clean up and consolidate redundant fields in RadixLinearAttention (#17660)

2026-01-27 10:37:58 -08:00

rocm_linear_utils.py

Aiter fp8 kv cache (#13147)

2025-12-08 16:39:53 -08:00

rotary_embedding.py

[Kernel] Add JIT rotary_embedding_kernel (#17934)

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

sampler.py

Make the functions in logits_processor.py and sampler.py more modular (#17885)

2026-01-28 16:24:23 -08:00

sparse_pooler.py

Implement BGE-M3 Sparse Embeddings in SGLang (#10869)

2025-10-22 13:46:16 -07:00

torchao_utils.py

Lazy import torchao (#17626)

2026-01-22 22:04:51 -08:00

vocab_parallel_embedding.py

Make sure we always disable symm memory without dp padding (#18129)

2026-02-04 19:58:28 +08:00