Jinzhen Lin ce96857fdd

[Kernel][Quantization][MoE] add marlin kernel support for turing (sm75) (#29901) ...

Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Co-authored-by: Michael Goin <mgoin64@gmail.com>

2025-12-16 14:35:28 -08:00

..

attention

[Performance][MLA][ROCm] Remove redundant D2D copy in deepseek (#27457)

2025-11-26 12:45:28 +08:00

core

…

cpu

[CPU] Support for Whisper (#30062)

2025-12-10 04:58:42 -08:00

cutlass_extensions

…

mamba/mamba_ssm

…

moe

[Kernel][Quantization][MoE] add marlin kernel support for turing (sm75) (#29901)

2025-12-16 14:35:28 -08:00

quantization

[Kernel][Quantization][MoE] add marlin kernel support for turing (sm75) (#29901)

2025-12-16 14:35:28 -08:00

quickreduce

…

rocm

Improve wvsplitK tile and balance heristics. (#29937)

2025-12-09 23:51:32 +00:00

sparse/cutlass

…

activation_kernels.cu

…

cache_kernels.cu

[Attention] Use sparse prefill kernel for fp8 kv-cache in DeepSeek-v3.2 (#27532)

2025-12-12 05:57:47 -08:00

cache.h

[Attention] Use sparse prefill kernel for fp8 kv-cache in DeepSeek-v3.2 (#27532)

2025-12-12 05:57:47 -08:00

cub_helpers.h

…

cuda_compat.h

…

cuda_utils_kernels.cu

…

cuda_utils.h

…

cuda_view.cu

Simplify from_blob usage in get_cuda_view_from_cpu_tensor (#29027)

2025-11-22 10:35:32 +00:00

cumem_allocator_compat.h

[Core][AMD] Migrate fully transparent sleep mode to ROCm platform (#12695)

2025-11-12 15:24:12 -08:00

cumem_allocator.cpp

[Core][AMD] Migrate fully transparent sleep mode to ROCm platform (#12695)

2025-11-12 15:24:12 -08:00

custom_all_reduce_test.cu

…

custom_all_reduce.cu

…

custom_all_reduce.cuh

…

custom_quickreduce.cu

…

dispatch_utils.h

[Performance] Fused blockwise quant RMS norm (#27883)

2025-12-07 16:38:04 +00:00

fused_qknorm_rope_kernel.cu

[ROCm][Bugfix] Fix compilation errors with fused_qknorm_rope_kernel.cu (#28682)

2025-11-14 11:06:01 -08:00

launch_bounds_utils.h

…

layernorm_kernels.cu

[kernel][perf] support uncontiguous input for rms_norm kernel (#28103)

2025-11-20 19:39:09 -08:00

layernorm_quant_kernels.cu

[Kernel] Optimize rms_norm kernel (#27931)

2025-11-11 18:02:23 +00:00

ops.h

[Kernel]Support W4A8 Grouped GEMM on Hopper (#29691)

2025-12-08 19:29:06 -08:00

permute_cols.cu

…

pos_encoding_kernels.cu

…

sampler.cu

[Bugfix][DSV32] Fix overflow in topk. (#30754)

2025-12-16 14:21:17 -08:00

torch_bindings.cpp

[Attention] Use sparse prefill kernel for fp8 kv-cache in DeepSeek-v3.2 (#27532)

2025-12-12 05:57:47 -08:00

type_convert.cuh

[ROCm] [Bugfix] Fix fused_qknorm_rope_kernel rocm compatibility (#28500)

2025-11-12 05:01:14 -08:00