| .. |
|
all_reduce
|
29eaded422
fix and re-enable custom all-reduce
|
hai 9 meses |
|
attention
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
backup
|
f8dfac6372
chore: attention refactor and upstream sync apr01 (#365)
|
hai 9 meses |
|
causal_conv1d
|
44b02e94cd
add forward kernels for causal depthwise conv1d
|
hai 8 meses |
|
cpu
|
fb23720c72
fix CPU build
|
hai 9 meses |
|
hadamard
|
5d288aa76c
feat: add fast hadamard transformation kernels (#232)
|
hai 11 meses |
|
moe
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
punica
|
c0aac15421
feat: S-LoRA support (#222)
|
hai 11 meses |
|
quantization
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
selective_scan
|
7fd3bd4bf2
add selective scan kernels
|
hai 8 meses |
|
activation_kernels.cu
|
3d6695cfbb
feat: add approximate gelu activation kernels (#370)
|
hai 9 meses |
|
cache.h
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
cache_kernels.cu
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
cuda_compat.h
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
cuda_utils.h
|
31c95011a6
feat: FP8 E5M2 KV Cache (#226)
|
hai 11 meses |
|
cuda_utils_kernels.cu
|
31c95011a6
feat: FP8 E5M2 KV Cache (#226)
|
hai 11 meses |
|
dispatch_utils.h
|
f8dfac6372
chore: attention refactor and upstream sync apr01 (#365)
|
hai 9 meses |
|
layernorm_kernels.cu
|
40f63268ee
disable new layernorm kernels for CUDA < 12.0
|
hai 9 meses |
|
ops.h
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
pos_encoding_kernels.cu
|
e702f587cf
feat: add batched RoPE kernels (#371)
|
hai 9 meses |
|
pybind.cpp
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
|
reduction.cuh
|
071269e406
feat: FP8 E4M3 KV Cache (#405)
|
hai 9 meses |
AlpinDale
7fd3bd4bf2
add selective scan kernels