whisper.cpp/ggml-cuda
Johannes Gäßler e57e95eb0d
CUDA: add FP32 FlashAttention vector kernel (llama/7188)
* CUDA: add FP32 FlashAttention vector kernel

* fixup! CUDA: add FP32 FlashAttention vector kernel

* fixup! fixup! CUDA: add FP32 FlashAttention vector kernel

* fixup! fixup! fixup! CUDA: add FP32 FlashAttention vector kernel
2024-05-14 19:16:29 +03:00
..
acc.cu
acc.cuh
arange.cu
arange.cuh
argsort.cu
argsort.cuh
binbcast.cu
binbcast.cuh
clamp.cu Introduction of CUDA Graphs to LLama.cpp (llama/6766) 2024-05-13 11:02:26 +03:00
clamp.cuh
common.cuh CUDA: add FP32 FlashAttention vector kernel (llama/7188) 2024-05-14 19:16:29 +03:00
concat.cu
concat.cuh
convert.cu Introduction of CUDA Graphs to LLama.cpp (llama/6766) 2024-05-13 11:02:26 +03:00
convert.cuh
cpy.cu Introduction of CUDA Graphs to LLama.cpp (llama/6766) 2024-05-13 11:02:26 +03:00
cpy.cuh Introduction of CUDA Graphs to LLama.cpp (llama/6766) 2024-05-13 11:02:26 +03:00
dequantize.cuh
diagmask.cu
diagmask.cuh
dmmv.cu
dmmv.cuh
fattn-common.cuh CUDA: add FP32 FlashAttention vector kernel (llama/7188) 2024-05-14 19:16:29 +03:00
fattn-vec-f16.cu CUDA: add FP32 FlashAttention vector kernel (llama/7188) 2024-05-14 19:16:29 +03:00
fattn-vec-f16.cuh CUDA: add FP32 FlashAttention vector kernel (llama/7188) 2024-05-14 19:16:29 +03:00
fattn-vec-f32.cu CUDA: add FP32 FlashAttention vector kernel (llama/7188) 2024-05-14 19:16:29 +03:00
fattn-vec-f32.cuh CUDA: add FP32 FlashAttention vector kernel (llama/7188) 2024-05-14 19:16:29 +03:00
fattn.cu CUDA: add FP32 FlashAttention vector kernel (llama/7188) 2024-05-14 19:16:29 +03:00
fattn.cuh ggml : add Flash Attention (llama/5021) 2024-05-13 11:02:26 +03:00
getrows.cu
getrows.cuh
im2col.cu
im2col.cuh
mmq.cu Introduction of CUDA Graphs to LLama.cpp (llama/6766) 2024-05-13 11:02:26 +03:00
mmq.cuh
mmvq.cu Introduction of CUDA Graphs to LLama.cpp (llama/6766) 2024-05-13 11:02:26 +03:00
mmvq.cuh
norm.cu
norm.cuh
pad.cu
pad.cuh
pool2d.cu
pool2d.cuh
quantize.cu
quantize.cuh
rope.cu
rope.cuh
scale.cu Introduction of CUDA Graphs to LLama.cpp (llama/6766) 2024-05-13 11:02:26 +03:00
scale.cuh
softmax.cu ggml : full ALiBi support (llama/7192) 2024-05-13 11:02:26 +03:00
softmax.cuh
sumrows.cu
sumrows.cuh
tsembd.cu
tsembd.cuh
unary.cu
unary.cuh
upscale.cu
upscale.cuh
vecdotq.cuh