| .. |
|
attention
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
quantization
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
activation_kernels.cu
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
cache.h
|
1aab8a7d6f
feat: speedup compilation times by 3x (#130)
|
1 year ago |
|
cache_kernels.cu
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
cuda_compat.h
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
cuda_utils.h
|
1aab8a7d6f
feat: speedup compilation times by 3x (#130)
|
1 year ago |
|
cuda_utils_kernels.cu
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
dispatch_utils.h
|
32844c1522
add GELU kernels and remove compile bloat
|
1 year ago |
|
layernorm_kernels.cu
|
7612f33afd
feat: fused add RMSNorm kernels (#125)
|
1 year ago |
|
ops.h
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
pos_encoding_kernels.cu
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
pybind.cpp
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
|
reduction.cuh
|
1334a833a4
feat: AMD ROCm support (#95)
|
1 year ago |
AlpinDale
1334a833a4
feat: AMD ROCm support (#95)