Michael Goin 53a5a0ce30

[Perf] Tunings for SM100 FP8 CUTLASS kernel (#18778) ...

Signed-off-by: mgoin <mgoin64@gmail.com>

2025-06-04 10:46:28 -07:00

..

attention

[NVIDIA] Add Cutlass MLA backend (#17625)

2025-06-03 21:40:26 -07:00

core

[Kernel] fp4 marlin kernel (#17687)

2025-05-10 19:58:49 -07:00

cpu

[Bugfix] Add half type support in reshape_and_cache_cpu_impl on x86 cpu platform (#18430)

2025-05-23 01:41:37 -07:00

cutlass_extensions

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

mamba

feat(rocm-support): support mamba2 on rocm (#18565)

2025-05-27 00:07:53 -07:00

moe

[Kernel] DeepEP dispatch-combine kernel integration (#18434)

2025-06-03 12:30:02 -07:00

prepare_inputs

…

quantization

[Perf] Tunings for SM100 FP8 CUTLASS kernel (#18778)

2025-06-04 10:46:28 -07:00

rocm

[ROCm][Kernel] Add gfx950 support for skinny gemms (#18010)

2025-05-31 07:40:05 -07:00

sparse/cutlass

fix CUDA_check redefinition in #17918 (#18287)

2025-05-19 13:42:35 -07:00

activation_kernels.cu

Modularize fused experts and integrate PPLX kernels (#15956)

2025-05-14 13:11:54 -07:00

cache_kernels.cu

…

cache.h

…

cuda_compat.h

…

cuda_utils_kernels.cu

…

cuda_utils.h

…

cuda_view.cu

…

cumem_allocator.cpp

…

custom_all_reduce_test.cu

…

custom_all_reduce.cu

…

custom_all_reduce.cuh

…

dispatch_utils.h

Modularize fused experts and integrate PPLX kernels (#15956)

2025-05-14 13:11:54 -07:00

layernorm_kernels.cu

…

layernorm_quant_kernels.cu

…

ops.h

[KERNEL] Sampler. CUDA kernel for applying repetition penalty (#18437)

2025-06-03 21:13:01 -07:00

permute_cols.cu

…

pos_encoding_kernels.cu

[Kernel] Have rotary embeddings support tensors (#18046)

2025-05-14 15:43:55 -07:00

sampler.cu

[KERNEL] Sampler. CUDA kernel for applying repetition penalty (#18437)

2025-06-03 21:13:01 -07:00

torch_bindings.cpp

[KERNEL] Sampler. CUDA kernel for applying repetition penalty (#18437)

2025-06-03 21:13:01 -07:00

type_convert.cuh

…