Wentao Ye 0ee6416f67

[Perf] Optimize group_topk kernel, 1.9% Throughput improvement, 2.1% TPOT improvemnt (#30159) ...

Signed-off-by: yewentao256 <zhyanwentao@126.com>

2025-12-08 19:44:01 -05:00

..

attention

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

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

core

…

cpu

[CPU][Perf] Add fast vectorized exp impl from Arm Optimized Routines (#30068)

2025-12-05 13:09:20 +00:00

cutlass_extensions

…

mamba/mamba_ssm

…

moe

[Perf] Optimize group_topk kernel, 1.9% Throughput improvement, 2.1% TPOT improvemnt (#30159)

2025-12-08 19:44:01 -05:00

quantization

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

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

quickreduce

…

rocm

…

sparse/cutlass

…

activation_kernels.cu

…

cache_kernels.cu

[Perf][Deepseek] optimize gather_and_maybe_dequant_cache kernel's perf for extremely long sequence (#28029)

2025-11-24 19:05:46 -07:00

cache.h

[Perf][Deepseek] optimize gather_and_maybe_dequant_cache kernel's perf for extremely long sequence (#28029)

2025-11-24 19:05:46 -07: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

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

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

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

2025-12-08 06:55:58 -08:00

type_convert.cuh

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

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