| .. |
|
template-instances
|
CUDA: optimize FA for GQA + large batches (llama/12014)
|
2025-02-27 08:55:36 +02:00 |
|
vendors
|
CUDA/HIP: Share the same unified memory allocation logic. (llama/12934)
|
2025-04-24 20:39:16 +03:00 |
|
CMakeLists.txt
|
build : fix build info on windows (llama/13239)
|
2025-05-07 15:39:32 +03: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
|
cuda: unary ops as float + de-duplicate (ggml/1130)
|
2025-03-08 15:13:01 +02:00 |
|
clamp.cuh
|
whisper : reorganize source code + improve CMake (#2256)
|
2024-06-26 19:34:09 +03:00 |
|
common.cuh
|
whisper: remove MSVC warnings pragmas (#3090)
|
2025-05-05 13:09:35 +02:00 |
|
concat.cu
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +03:00 |
|
concat.cuh
|
whisper : reorganize source code + improve CMake (#2256)
|
2024-06-26 19:34:09 +03:00 |
|
conv-transpose-1d.cu
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +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: fix non-cont. inputs for batched mat mul (llama/13155)
|
2025-05-01 13:29:02 +03:00 |
|
convert.cuh
|
CUDA: fix non-cont. inputs for batched mat mul (llama/13155)
|
2025-05-01 13:29:02 +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
|
musa: fix build warning (llama/13129)
|
2025-05-01 13:29:02 +03:00 |
|
cpy.cuh
|
ggml: Re-enable CUDA graphs in presence of CONT and DUP nodes (llama/12970)
|
2025-04-24 20:39:16 +03: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
|
musa: fix compilation warnings in mp_22/31 (llama/12780)
|
2025-04-24 20:39:16 +03:00 |
|
fattn-mma-f16.cuh
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +03:00 |
|
fattn-tile-f16.cu
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +03:00 |
|
fattn-tile-f16.cuh
|
whisper : reorganize source code + improve CMake (#2256)
|
2024-06-26 19:34:09 +03:00 |
|
fattn-tile-f32.cu
|
musa: fix compilation warnings in mp_22/31 (llama/12780)
|
2025-04-24 20:39:16 +03:00 |
|
fattn-tile-f32.cuh
|
whisper : reorganize source code + improve CMake (#2256)
|
2024-06-26 19:34:09 +03:00 |
|
fattn-vec-f16.cuh
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +03:00 |
|
fattn-vec-f32.cuh
|
musa: fix compilation warnings in mp_22/31 (llama/12780)
|
2025-04-24 20:39:16 +03:00 |
|
fattn-wmma-f16.cu
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +03: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: Prefer vector flash decoding kernel for Gemma models (llama/12738)
|
2025-04-24 20:39:16 +03:00 |
|
fattn.cuh
|
whisper : reorganize source code + improve CMake (#2256)
|
2024-06-26 19:34:09 +03:00 |
|
getrows.cu
|
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (llama/13199)
|
2025-05-01 13:29:02 +03:00 |
|
getrows.cuh
|
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (llama/13199)
|
2025-05-01 13:29:02 +03:00 |
|
ggml-cuda.cu
|
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (llama/13199)
|
2025-05-01 13:29:02 +03: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
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +03:00 |
|
mmq.cu
|
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (llama/13199)
|
2025-05-01 13:29:02 +03:00 |
|
mmq.cuh
|
CUDA: fix race condition in MMQ ids_dst (llama/13294)
|
2025-05-07 21:00:32 +03:00 |
|
mmv.cu
|
CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (llama/13014)
|
2025-04-24 20:39:16 +03:00 |
|
mmv.cuh
|
CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (llama/13014)
|
2025-04-24 20:39:16 +03:00 |
|
mmvq.cu
|
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (llama/13199)
|
2025-05-01 13:29:02 +03:00 |
|
mmvq.cuh
|
CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (llama/13014)
|
2025-04-24 20:39:16 +03:00 |
|
norm.cu
|
llama: Add support for RWKV v7 architecture (llama/12412)
|
2025-03-27 11:06:03 +02:00 |
|
norm.cuh
|
llama: Add support for RWKV v7 architecture (llama/12412)
|
2025-03-27 11:06:03 +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
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +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: batched+noncont MMQ, refactor bs>1 MoE code (llama/13199)
|
2025-05-01 13:29:02 +03:00 |
|
quantize.cuh
|
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (llama/13199)
|
2025-05-01 13:29:02 +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 |
|
ssm-conv.cu
|
fix MUSA compiler warning (llama/12704)
|
2025-04-24 20:39:16 +03:00 |
|
ssm-conv.cuh
|
ggml : faster ssm scan (llama/10558)
|
2025-04-02 15:51:57 +03:00 |
|
ssm-scan.cu
|
fix MUSA compiler warning (llama/12704)
|
2025-04-24 20:39:16 +03:00 |
|
ssm-scan.cuh
|
ggml : faster ssm scan (llama/10558)
|
2025-04-02 15:51:57 +03: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: unary ops as float + de-duplicate (ggml/1130)
|
2025-03-08 15:13:01 +02:00 |
|
unary.cuh
|
cuda/cpu: Increase support for fp16 unary operations (ggml/1125)
|
2025-03-08 15:13:01 +02:00 |
|
upscale.cu
|
musa: fix all warnings, re-enable `-DLLAMA_FATAL_WARNINGS=ON` in ci and update doc (llama/12611)
|
2025-03-31 14:56:53 +03:00 |
|
upscale.cuh
|
whisper : reorganize source code + improve CMake (#2256)
|
2024-06-26 19:34:09 +03:00 |
|
vecdotq.cuh
|
CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (llama/13014)
|
2025-04-24 20:39:16 +03:00 |
|
wkv.cu
|
llama: Add support for RWKV v7 architecture (llama/12412)
|
2025-03-27 11:06:03 +02:00 |
|
wkv.cuh
|
llama: Add support for RWKV v7 architecture (llama/12412)
|
2025-03-27 11:06:03 +02:00 |