Wentao Ye f21f5ea38c

[Refactor] Small refactor for group topk (#30562) ...

Signed-off-by: yewentao256 <zhyanwentao@126.com>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>

2025-12-16 14:50:59 -05: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

[Refactor] Small refactor for group topk (#30562)

2025-12-16 14:50:59 -05:00

quantization

[Refactor] Reduce duplicate code in per_token_group_quant cuda kernels (#30496)

2025-12-12 16:45:23 -05: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

[DeepSeek v3.2] Make top-k work for any logit values. (#27568)

2025-12-08 06:55:58 -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