| .. |
|
all_reduce
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
attention
|
3bb0f07461
chore: rename `task_handler` to `worker` (#985)
|
il y a 2 mois |
|
backup
|
3bb0f07461
chore: rename `task_handler` to `worker` (#985)
|
il y a 2 mois |
|
core
|
f1ea7711bd
core: do not compile ScalarType for torch < 2.4.0 (#938)
|
il y a 2 mois |
|
cpu
|
a113309876
kernel: add meta functions for ops to prevent graph breaks (#1019)
|
il y a 2 mois |
|
cutlass_extensions
|
93bc863591
feat: Machete Kernels for Hopper GPUs (#842)
|
il y a 3 mois |
|
hadamard
|
5d288aa76c
feat: add fast hadamard transformation kernels (#232)
|
il y a 1 an |
|
mamba
|
0256ed236b
feat: windows support (#790)
|
il y a 4 mois |
|
moe
|
9f3e7c86e2
feat: add fused Marlin MoE kernel (#934)
|
il y a 2 mois |
|
prepare_inputs
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
punica
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
quantization
|
a113309876
kernel: add meta functions for ops to prevent graph breaks (#1019)
|
il y a 2 mois |
|
sampling
|
bfc8988116
feat: add cuda sampling kernels for top_k and top_p (#828)
|
il y a 3 mois |
|
activation_kernels.cu
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
cache.h
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
cache_kernels.cu
|
0256ed236b
feat: windows support (#790)
|
il y a 4 mois |
|
cuda_compat.h
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
cuda_utils.h
|
93bc863591
feat: Machete Kernels for Hopper GPUs (#842)
|
il y a 3 mois |
|
cuda_utils_kernels.cu
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
dispatch_utils.h
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
layernorm_kernels.cu
|
e14223dce5
kernel: use `cub::BlockReduce` instead of custom impl (#895)
|
il y a 2 mois |
|
ops.h
|
93bc863591
feat: Machete Kernels for Hopper GPUs (#842)
|
il y a 3 mois |
|
permute_cols.cu
|
93bc863591
feat: Machete Kernels for Hopper GPUs (#842)
|
il y a 3 mois |
|
pos_encoding_kernels.cu
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
il y a 6 mois |
|
torch_bindings.cpp
|
a113309876
kernel: add meta functions for ops to prevent graph breaks (#1019)
|
il y a 2 mois |
AlpinDale
a113309876
kernel: add meta functions for ops to prevent graph breaks (#1019)