| .. |
|
all_reduce
|
239a8cae25
torch.compile: register all-reduce operations as custom ops (#1050)
|
il y a 1 semaine |
|
attention
|
3bb0f07461
chore: rename `task_handler` to `worker` (#985)
|
il y a 2 semaines |
|
backup
|
3bb0f07461
chore: rename `task_handler` to `worker` (#985)
|
il y a 2 semaines |
|
core
|
f1ea7711bd
core: do not compile ScalarType for torch < 2.4.0 (#938)
|
il y a 2 semaines |
|
cpu
|
8976805f90
kernel: asymmetric AQ AZP quantization kernels (#1048)
|
il y a 1 semaine |
|
cutlass_extensions
|
93bc863591
feat: Machete Kernels for Hopper GPUs (#842)
|
il y a 1 mois |
|
hadamard
|
5d288aa76c
feat: add fast hadamard transformation kernels (#232)
|
il y a 11 mois |
|
mamba
|
9bdf8d5bfa
mamba: enable continuous batching for mamba kernels (#1055)
|
il y a 1 semaine |
|
moe
|
9f3e7c86e2
feat: add fused Marlin MoE kernel (#934)
|
il y a 2 semaines |
|
prepare_inputs
|
1390915778
multi-step: add support for flashinfer attention backend (#1033)
|
il y a 1 semaine |
|
punica
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
quantization
|
8976805f90
kernel: asymmetric AQ AZP quantization kernels (#1048)
|
il y a 1 semaine |
|
rocm
|
61aed092a5
rocm: add support for FP8 KV cache in the custom paged attention kkernels (#1066)
|
il y a 6 jours |
|
sampling
|
bfc8988116
feat: add cuda sampling kernels for top_k and top_p (#828)
|
il y a 1 mois |
|
activation_kernels.cu
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
cache.h
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
cache_kernels.cu
|
0256ed236b
feat: windows support (#790)
|
il y a 2 mois |
|
cuda_compat.h
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
cuda_utils.h
|
93bc863591
feat: Machete Kernels for Hopper GPUs (#842)
|
il y a 1 mois |
|
cuda_utils_kernels.cu
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
dispatch_utils.h
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
layernorm_kernels.cu
|
e14223dce5
kernel: use `cub::BlockReduce` instead of custom impl (#895)
|
il y a 3 semaines |
|
ops.h
|
525edc1283
build: fix compilation for causal_conv1d_fwd kernel signature (#1057)
|
il y a 1 semaine |
|
permute_cols.cu
|
93bc863591
feat: Machete Kernels for Hopper GPUs (#842)
|
il y a 1 mois |
|
pos_encoding_kernels.cu
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
torch_bindings.cpp
|
9bdf8d5bfa
mamba: enable continuous batching for mamba kernels (#1055)
|
il y a 1 semaine |
AlpinDale
61aed092a5
rocm: add support for FP8 KV cache in the custom paged attention kkernels (#1066)