Xiaoyu Zhang c29394e3c8 [kernel slimming] Move fast_hadamard_transform to jit_kernel (#18475)

2026-02-14 23:06:21 +08:00

..

attention

[kernel slimming] Move fast_hadamard_transform to jit_kernel (#18475)

2026-02-14 23:06:21 +08:00

deep_gemm_wrapper

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

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

moe

[VLM][LLM] Optimize fused_moe triton kernel tma (#18782)

2026-02-14 14:35:26 +08:00

quantization

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

2026-02-13 22:29:22 +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

refactor context parallel state (#17213)

2026-02-13 23:18:17 +08:00

communicator.py

refactor context parallel state (#17213)

2026-02-13 23:18:17 +08:00

dp_attention.py

refactor context parallel state (#17213)

2026-02-13 23:18:17 +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

…

multimodal.py

…

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

[Perf] refactor piecewise cuda graph support of Qwen3-Next (#17613)

2026-02-14 09:30:50 +08:00

rocm_linear_utils.py

Aiter fp8 kv cache (#13147)

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

rotary_embedding.py

feat: Support mrope_section with rope_type: "yarn" (#13313)

2026-02-14 22:51:44 +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

…

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