| .. |
|
all_reduce
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
attention
|
f831fd8312
rocm: fix compile issues with rocm 6.2 (#893)
|
il y a 1 mois |
|
backup
|
f8dfac6372
chore: attention refactor and upstream sync apr01 (#365)
|
il y a 9 mois |
|
core
|
f1ea7711bd
core: do not compile ScalarType for torch < 2.4.0 (#938)
|
il y a 1 mois |
|
cpu
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
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
|
0256ed236b
feat: windows support (#790)
|
il y a 2 mois |
|
moe
|
9f3e7c86e2
feat: add fused Marlin MoE kernel (#934)
|
il y a 1 mois |
|
prepare_inputs
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
punica
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 4 mois |
|
quantization
|
2a60b8f8c9
kernel: do not compile machete for cuda 11 and below (#901)
|
il y a 1 mois |
|
sampling
|
427f6b32fb
kernels for rejection sampling
|
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 1 mois |
|
ops.h
|
427f6b32fb
kernels for rejection sampling
|
il y a 1 mois |
|
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
|
427f6b32fb
kernels for rejection sampling
|
il y a 1 mois |
AlpinDale
427f6b32fb
kernels for rejection sampling