3 people e2290b155a Port KV Compression V2 from deepseek_v4_dev (#24890) ...

Co-authored-by: Cheng Wan <chwan@rice.edu>
Co-authored-by: DarkSharpness <2040703891@qq.com>

2026-05-13 22:40:38 +08:00

..

deepseek_v4

Port KV Compression V2 from deepseek_v4_dev (#24890)

2026-05-13 22:40:38 +08:00

distributed

[Feature][JIT Kernel] Fused TP QK norm For Minimax (#20673)

2026-04-13 20:29:47 +08:00

impl

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

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

atomic.cuh

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

cta.cuh

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

ffi.h

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

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

math.cuh

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

runtime.cuh

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

scalar_type.hpp

[Kernel] Migrate GPTQ-Marlin GEMM kernel to JIT (#18067)

2026-02-06 08:31:42 +08:00

source_location.h

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

tensor.h

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

tile.cuh

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

type.cuh

[jit_kernel] Migrate cast (downcast_fp8) from sgl-kernel AOT to JIT (#19103)

2026-03-27 13:21:44 +08:00

utils.cuh

Deepseek V4 (#23882)

2026-05-07 18:32:21 -07:00

utils.h

[Docs] Add docstrings to JIT kernel include headers (#19770)

2026-03-07 20:48:00 +08:00

vec.cuh

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

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

warp.cuh

Deepseek V4 (#23882)

2026-05-07 18:32:21 -07:00