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