Gregory Shtrasberg aabeb2688f

[ROCm][Quantization][Kernel] Using HIP FP8 header (#12593)

2025-02-25 00:39:59 -08:00

..

attention

…

core

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

cpu

…

cutlass_extensions

[Kernel][Bugfix] Refactor and Fix CUTLASS 2:4 Sparse Kernels (#13198)

2025-02-14 00:01:14 +00:00

mamba

…

moe

Optimize moe_align_block_size for deepseek_v3 (#12850)

2025-02-13 18:43:37 -05:00

prepare_inputs

…

quantization

[ROCm][Quantization][Kernel] Using HIP FP8 header (#12593)

2025-02-25 00:39:59 -08:00

rocm

[ROCm] MI300A compile targets deprecation (#13560)

2025-02-19 23:05:00 -08:00

sparse/cutlass

[Kernel][Bugfix] Refactor and Fix CUTLASS 2:4 Sparse Kernels (#13198)

2025-02-14 00:01:14 +00:00

activation_kernels.cu

…

cache_kernels.cu

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

cache.h

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

cuda_compat.h

…

cuda_utils_kernels.cu

[NVIDIA] Support nvfp4 quantization (#12784)

2025-02-12 19:51:51 -08:00

cuda_utils.h

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

cumem_allocator.cpp

[core] improve error handling when wake up from sleep mode (#12981)

2025-02-10 09:38:57 +08:00

custom_all_reduce_test.cu

…

custom_all_reduce.cu

…

custom_all_reduce.cuh

…

dispatch_utils.h

…

layernorm_kernels.cu

…

layernorm_quant_kernels.cu

…

ops.h

[NVIDIA] Support nvfp4 cutlass gemm (#13571)

2025-02-22 05:24:05 -08:00

permute_cols.cu

…

pos_encoding_kernels.cu

[Kernel] Make rotary_embedding ops more flexible with input shape (#12777)

2025-02-06 08:46:13 -08:00

torch_bindings.cpp

[NVIDIA] Support nvfp4 cutlass gemm (#13571)

2025-02-22 05:24:05 -08:00

type_convert.cuh

…