..
template-instances
CUDA: optimize FA for GQA + large batches (llama/12014)
2025-02-27 08:55:36 +02:00
vendors
CUDA/HIP: add support for selectable warp size to mmv (llama/11519)
2025-02-03 22:00:57 +02:00
CMakeLists.txt
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
acc.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
acc.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
arange.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
arange.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
argmax.cu
cuda : optimize argmax (llama/10441)
2024-12-08 20:14:35 +02:00
argmax.cuh
ggml/ex: calculate accuracy in graph, adapt MNIST (ggml/980)
2024-10-05 15:23:51 +03:00
argsort.cu
ggml : reduce hash table reset cost (llama/8698)
2024-08-08 22:48:46 +03:00
argsort.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
binbcast.cu
Support pure float16 add/sub/mul/div operations in the CUDA (and CPU) backend (ggml/1121)
2025-02-27 08:55:36 +02:00
binbcast.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-24 19:45:08 +03:00
clamp.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
clamp.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
common.cuh
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
concat.cu
fix: add missing msg in static_assert (llama/11143)
2025-01-14 10:38:01 +02:00
concat.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
conv-transpose-1d.cu
feat: cuda implementation for `ggml_conv_transpose_1d` (ggml/854)
2024-07-08 14:53:55 +03:00
conv-transpose-1d.cuh
feat: cuda implementation for `ggml_conv_transpose_1d` (ggml/854)
2024-07-08 14:53:55 +03:00
convert.cu
CUDA: use arch list for compatibility check (llama/11775)
2025-02-27 08:55:36 +02:00
convert.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
count-equal.cu
ggml: fix zero division in ‘dne’ calculation in CUDA COUNT_EQUAL operator when ‘ne’ is small ( #10213 )
2024-11-15 15:21:04 +02:00
count-equal.cuh
ggml/ex: calculate accuracy in graph, adapt MNIST (ggml/980)
2024-10-05 15:23:51 +03:00
cp-async.cuh
CUDA: optimize FA for GQA + large batches (llama/12014)
2025-02-27 08:55:36 +02:00
cpy.cu
cuda: Add Q5_1, Q5_0, Q4_1 and Q4_0 to F32 conversion support. (llama/12000)
2025-02-27 08:55:36 +02:00
cpy.cuh
increase cuda_cpy block size (ggml/996)
2024-11-01 10:19:05 +02:00
cross-entropy-loss.cu
MUSA: support ARM64 and enable dp4a .etc (llama/11843)
2025-02-27 08:55:36 +02:00
cross-entropy-loss.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-24 19:45:08 +03:00
dequantize.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
diagmask.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
diagmask.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
fattn-common.cuh
CUDA: optimize FA for GQA + large batches (llama/12014)
2025-02-27 08:55:36 +02:00
fattn-mma-f16.cuh
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
fattn-tile-f16.cu
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
fattn-tile-f16.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
fattn-tile-f32.cu
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
fattn-tile-f32.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
fattn-vec-f16.cuh
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
fattn-vec-f32.cuh
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
fattn-wmma-f16.cu
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
fattn-wmma-f16.cuh
CUDA: use mma PTX instructions for FlashAttention (llama/11583)
2025-02-03 22:00:57 +02:00
fattn.cu
CUDA: optimize FA for GQA + large batches (llama/12014)
2025-02-27 08:55:36 +02:00
fattn.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
getrows.cu
CUDA: backwards pass for misc. ops, add tests (llama/11257)
2025-02-03 22:00:57 +02:00
getrows.cuh
CUDA: backwards pass for misc. ops, add tests (llama/11257)
2025-02-03 22:00:57 +02:00
ggml-cuda.cu
CUDA: app option to compile without FlashAttention (llama/12025)
2025-02-27 08:55:36 +02:00
gla.cu
llama: add support for QRWKV6 model architecture (llama/11001)
2025-01-14 10:38:01 +02:00
gla.cuh
llama: add support for QRWKV6 model architecture (llama/11001)
2025-01-14 10:38:01 +02:00
im2col.cu
CUDA: fix 1D im2col, add tests (ggml/993)
2024-11-01 10:19:05 +02:00
im2col.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
mma.cuh
CUDA: optimize FA for GQA + large batches (llama/12014)
2025-02-27 08:55:36 +02:00
mmq.cu
HIP: Remove GCN from list of devices that avoid MMQ (llama/11831)
2025-02-27 08:55:36 +02:00
mmq.cuh
CUDA: use async data loading for FlashAttention (llama/11894)
2025-02-27 08:55:36 +02:00
mmv.cu
CUDA: support for mat. mul. with ne03 != ne13 (llama/11656)
2025-02-27 08:55:36 +02:00
mmv.cuh
CUDA: remove DMMV, consolidate F16 mult mat vec (llama/10318)
2024-11-20 21:00:08 +02:00
mmvq.cu
rocBLAS: Avoid fp32->fp16->fp32 conversion on cdna (llama/11356)
2025-02-03 22:00:57 +02:00
mmvq.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
norm.cu
CUDA: non-contiguous (RMS) norm support (llama/11659)
2025-02-27 08:55:36 +02:00
norm.cuh
CUDA: backwards pass for misc. ops, add tests (llama/11257)
2025-02-03 22:00:57 +02:00
opt-step-adamw.cu
ggml: new optimization interface (ggml/988)
2024-11-20 21:00:08 +02:00
opt-step-adamw.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-24 19:45:08 +03:00
out-prod.cu
CPU/CUDA: fix (GQA) mul mat back, add CUDA support (llama/11380)
2025-02-03 22:00:57 +02:00
out-prod.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-24 19:45:08 +03:00
pad.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
pad.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
pool2d.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
pool2d.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
quantize.cu
cuda : optimize argmax (llama/10441)
2024-12-08 20:14:35 +02:00
quantize.cuh
CUDA: optimize and refactor MMQ (llama/8416)
2024-08-08 22:48:46 +03:00
rope.cu
CUDA: backwards pass for misc. ops, add tests (llama/11257)
2025-02-03 22:00:57 +02:00
rope.cuh
RoPE: fix back, CUDA support for back + noncont. (llama/11240)
2025-02-03 22:00:57 +02:00
scale.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
scale.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
softmax.cu
HIP: fix flash_attn_stream_k_fixup warning (llama/11604)
2025-02-03 22:00:57 +02:00
softmax.cuh
CUDA: backwards pass for misc. ops, add tests (llama/11257)
2025-02-03 22:00:57 +02:00
sum.cu
CUDA: fix CUDART_VERSION checks (llama/11821)
2025-02-27 08:55:36 +02:00
sum.cuh
tests: add gradient tests for all backends (ggml/932)
2024-09-24 19:45:08 +03:00
sumrows.cu
feat: ref. cross entropy, add CUDA, fix grad test (ggml/929)
2024-08-28 13:22:20 +03:00
sumrows.cuh
feat: ref. cross entropy, add CUDA, fix grad test (ggml/929)
2024-08-28 13:22:20 +03:00
tsembd.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
tsembd.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
unary.cu
CUDA: backwards pass for misc. ops, add tests (llama/11257)
2025-02-03 22:00:57 +02:00
unary.cuh
CUDA: backwards pass for misc. ops, add tests (llama/11257)
2025-02-03 22:00:57 +02:00
upscale.cu
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
upscale.cuh
whisper : reorganize source code + improve CMake ( #2256 )
2024-06-26 19:34:09 +03:00
vecdotq.cuh
CUDA: MMQ code deduplication + iquant support (llama/8495)
2024-08-08 22:48:46 +03:00
wkv6.cu
llama: add support for QRWKV6 model architecture (llama/11001)
2025-01-14 10:38:01 +02:00
wkv6.cuh
Optimize RWKV6 Operator Naming and Implement Multi-core CPU/ SYCL Acceleration (llama/10133)
2024-11-15 15:21:04 +02:00