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 年間 前