Benji Beck 1f491aa0c8

Vectorize RMS norm variance using vectorize_read_with_alignment (#26234) ...

Signed-off-by: Benji Beck <benjibeck@meta.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>

2025-10-15 11:54:41 -07:00

..

attention

…

core

…

cpu

…

cutlass_extensions

…

mamba/mamba_ssm

…

moe

[unrevert] Add batch invariant kernel override for FlashInfer backend [2/n] (#26373)

2025-10-13 10:24:53 -04:00

quantization

…

quickreduce

[Bugfix][Rocm] fix qr error when different inp shape (#25892)

2025-10-13 10:04:21 -07:00

rocm

…

sparse/cutlass

…

activation_kernels.cu

…

cache_kernels.cu

…

cache.h

…

cub_helpers.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

…

custom_quickreduce.cu

…

dispatch_utils.h

…

launch_bounds_utils.h

…

layernorm_kernels.cu

Vectorize RMS norm variance using vectorize_read_with_alignment (#26234)

2025-10-15 11:54:41 -07:00

layernorm_quant_kernels.cu

Vectorize RMS norm variance using vectorize_read_with_alignment (#26234)

2025-10-15 11:54:41 -07:00

ops.h

…

permute_cols.cu

…

pos_encoding_kernels.cu

…

sampler.cu

…

torch_bindings.cpp

…

type_convert.cuh

…