| .. |
|
all_reduce
|
6305e6f3f2
fix: no repeated IPC registration (#227)
|
il y a 11 mois |
|
attention
|
8e1cd54497
fix: do not include fp8 for rocm (#271)
|
il y a 10 mois |
|
hadamard
|
5d288aa76c
feat: add fast hadamard transformation kernels (#232)
|
il y a 11 mois |
|
moe
|
7d6ba53602
feat: fused top-k kernels for MoE (#273)
|
il y a 10 mois |
|
punica
|
c0aac15421
feat: S-LoRA support (#222)
|
il y a 11 mois |
|
quantization
|
8e1cd54497
fix: do not include fp8 for rocm (#271)
|
il y a 10 mois |
|
activation_kernels.cu
|
e31c6f0b45
feat: refactor modeling logic and support more models (#274)
|
il y a 10 mois |
|
cache.h
|
31c95011a6
feat: FP8 E5M2 KV Cache (#226)
|
il y a 11 mois |
|
cache_kernels.cu
|
8e1cd54497
fix: do not include fp8 for rocm (#271)
|
il y a 10 mois |
|
cuda_compat.h
|
8fa608aeb7
feat: replace Ray with NCCL for control plane comms (#221)
|
il y a 11 mois |
|
cuda_utils.h
|
31c95011a6
feat: FP8 E5M2 KV Cache (#226)
|
il y a 11 mois |
|
cuda_utils_kernels.cu
|
31c95011a6
feat: FP8 E5M2 KV Cache (#226)
|
il y a 11 mois |
|
dispatch_utils.h
|
d9b65e6c5f
feat: DeepSeek MoE support (#237)
|
il y a 11 mois |
|
layernorm_kernels.cu
|
8fa608aeb7
feat: replace Ray with NCCL for control plane comms (#221)
|
il y a 11 mois |
|
ops.h
|
e31c6f0b45
feat: refactor modeling logic and support more models (#274)
|
il y a 10 mois |
|
pos_encoding_kernels.cu
|
8fa608aeb7
feat: replace Ray with NCCL for control plane comms (#221)
|
il y a 11 mois |
|
pybind.cpp
|
e31c6f0b45
feat: refactor modeling logic and support more models (#274)
|
il y a 10 mois |
|
reduction.cuh
|
2755a48d51
merge dev branch into main (#153)
|
il y a 1 an |
AlpinDale
e31c6f0b45
feat: refactor modeling logic and support more models (#274)