Aman Gupta 27208bf657 CUDA: add bf16 and f32 support to cublas_mul_mat_batched (#14361) ...

* CUDA: add bf16 and f32 support to cublas_mul_mat_batched

* Review: add type traits and make function more generic

* Review: make check more explicit, add back comments, and fix formatting

* Review: fix formatting, remove useless type conversion, fix naming for bools

2025-06-29 01:30:53 +08:00

..

template-instances

CUDA: FA support for Deepseek (Ampere or newer) (#13306)

2025-05-09 13:34:58 +02:00

vendors

CUDA/HIP: Share the same unified memory allocation logic. (#12934)

2025-04-15 11:20:38 +02:00

acc.cu

llama/ggml: add LLM training support (#10544)

2025-05-12 14:44:49 +02:00

acc.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

arange.cu

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

arange.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

argmax.cu

cuda : optimize argmax (#10441)

2024-11-21 18:18:50 +01:00

argmax.cuh

ggml/ex: calculate accuracy in graph, adapt MNIST (ggml/980)

2024-10-03 21:17:26 +03:00

argsort.cu

ggml : reduce hash table reset cost (#8698)

2024-07-27 04:41:55 +02:00

argsort.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

binbcast.cu

Support pure float16 add/sub/mul/div operations in the CUDA (and CPU) backend (ggml/1121)

2025-03-03 18:18:11 +02:00

binbcast.cuh

ggml/examples: add backend support for numerical optimization (ggml/949)

2024-09-20 21:15:05 +03:00

clamp.cu

cuda: unary ops as float + de-duplicate (ggml/1130)

2025-03-03 18:18:11 +02:00

clamp.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

CMakeLists.txt

CUDA: FA support for Deepseek (Ampere or newer) (#13306)

2025-05-09 13:34:58 +02:00

common.cuh

musa: enable fp16 mma (all) and cublas on qy2 (#13842)

2025-06-26 12:11:59 +08:00

concat.cu

musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)

2025-03-30 10:59:38 +02:00

concat.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

conv2d-dw.cu

CUDA: add conv_2d_dw (#14265)

2025-06-20 09:50:24 +08:00

conv2d-dw.cuh

CUDA: add conv_2d_dw (#14265)

2025-06-20 09:50:24 +08:00

conv2d-transpose.cu

CUDA: add conv_2d_transpose (#14287)

2025-06-20 22:48:24 +08:00

conv2d-transpose.cuh

CUDA: add conv_2d_transpose (#14287)

2025-06-20 22:48:24 +08:00

conv-transpose-1d.cu

musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)

2025-03-30 10:59:38 +02:00

conv-transpose-1d.cuh

feat: cuda implementation for ggml_conv_transpose_1d (ggml/854)

2024-07-08 12:23:00 +03:00

convert.cu

CUDA: add bf16 and f32 support to cublas_mul_mat_batched (#14361)

2025-06-29 01:30:53 +08:00

convert.cuh

CUDA: add bf16 and f32 support to cublas_mul_mat_batched (#14361)

2025-06-29 01:30:53 +08:00

count-equal.cu

ggml: fix zero division in ‘dne’ calculation in CUDA COUNT_EQUAL operator when ‘ne’ is small (#10213)

2024-11-09 08:35:46 +01:00

count-equal.cuh

ggml/ex: calculate accuracy in graph, adapt MNIST (ggml/980)

2024-10-03 21:17:26 +03:00

cp-async.cuh

CUDA: FA support for Deepseek (Ampere or newer) (#13306)

2025-05-09 13:34:58 +02:00

cpy.cu

musa: Upgrade MUSA SDK version to rc4.0.1 and use mudnn::Unary::IDENTITY op to accelerate D2D memory copy (#13647)

2025-05-21 09:58:49 +08:00

cpy.cuh

ggml: Re-enable CUDA graphs in presence of CONT and DUP nodes (#12970)

2025-04-17 15:19:42 +02:00

cross-entropy-loss.cu

MUSA: support ARM64 and enable dp4a .etc (#11843)

2025-02-21 09:46:23 +02:00

cross-entropy-loss.cuh

ggml/examples: add backend support for numerical optimization (ggml/949)

2024-09-20 21:15:05 +03:00

dequantize.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

diagmask.cu

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

diagmask.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

fattn-common.cuh

CUDA: fix FA tg at long context for CC >= 8.9 (#13852)

2025-05-28 13:33:37 +02:00

fattn-mma-f16.cuh

CUDA: fix FTZ in FA for Gemma 3 (#13991)

2025-06-04 08:57:05 +02:00

fattn-tile-f16.cu

CUDA: FA support for Deepseek (Ampere or newer) (#13306)

2025-05-09 13:34:58 +02:00

fattn-tile-f16.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

fattn-tile-f32.cu

CUDA: FA support for Deepseek (Ampere or newer) (#13306)

2025-05-09 13:34:58 +02:00

fattn-tile-f32.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

fattn-vec-f16.cuh

CUDA: fix race condition in FA vector kernels (#13742)

2025-05-24 11:46:19 +02:00

fattn-vec-f32.cuh

CUDA: fix race condition in FA vector kernels (#13742)

2025-05-24 11:46:19 +02:00

fattn-wmma-f16.cu

musa: enable fp16 mma (all) and cublas on qy2 (#13842)

2025-06-26 12:11:59 +08:00

fattn-wmma-f16.cuh

CUDA: use mma PTX instructions for FlashAttention (#11583)

2025-02-02 19:31:09 +01:00

fattn.cu

CUDA: faster Deepseek FA, add Turing support (#13435)

2025-05-14 16:08:20 +02:00

fattn.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

getrows.cu

CUDA: fix crash on large batch size for MoE models (#13384)

2025-05-09 12:14:04 +02:00

getrows.cuh

CUDA: batched+noncont MMQ, refactor bs>1 MoE code (#13199)

2025-04-30 23:12:59 +02:00

ggml-cuda.cu

CUDA: add bf16 and f32 support to cublas_mul_mat_batched (#14361)

2025-06-29 01:30:53 +08:00

gla.cu

llama: add support for QRWKV6 model architecture (#11001)

2025-01-10 09:58:08 +08:00

gla.cuh

llama: add support for QRWKV6 model architecture (#11001)

2025-01-10 09:58:08 +08:00

im2col.cu

CUDA: fix 1D im2col, add tests (ggml/993)

2024-10-23 16:50:02 +03:00

im2col.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

mean.cu

CUDA: add mean operation (#14313)

2025-06-22 12:39:54 +08:00

mean.cuh

CUDA: add mean operation (#14313)

2025-06-22 12:39:54 +08:00

mma.cuh

musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)

2025-03-30 10:59:38 +02:00

mmq.cu

CUDA: fix crash on large batch size for quant. MoE (#13537)

2025-05-14 16:41:02 +02:00

mmq.cuh

cuda : remove nrows_x in mul_mat_q_process_tile (#13325)

2025-05-07 09:48:23 +02:00

mmv.cu

CUDA/HIP: optimize mmv paths taken for HIP devices (#14324)

2025-06-24 01:12:56 +02:00

mmv.cuh

CUDA: mul_mat_v support for batch sizes > 1 (#14262)

2025-06-23 13:11:31 +02:00

mmvq.cu

CUDA: fix crash with partial offloading of MoE (#13439)

2025-05-11 16:09:33 +02:00

mmvq.cuh

CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (#13014)

2025-04-22 21:27:40 +02:00

norm.cu

llama: Add support for RWKV v7 architecture (#12412)

2025-03-18 07:27:50 +08:00

norm.cuh

llama: Add support for RWKV v7 architecture (#12412)

2025-03-18 07:27:50 +08:00

opt-step-adamw.cu

ggml: new optimization interface (ggml/988)

2024-11-17 08:30:29 +02:00

opt-step-adamw.cuh

ggml/examples: add backend support for numerical optimization (ggml/949)

2024-09-20 21:15:05 +03:00

out-prod.cu

CPU/CUDA: fix (GQA) mul mat back, add CUDA support (#11380)

2025-01-24 12:38:31 +01:00

out-prod.cuh

ggml/examples: add backend support for numerical optimization (ggml/949)

2024-09-20 21:15:05 +03:00

pad.cu

musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)

2025-03-30 10:59:38 +02:00

pad.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

pool2d.cu

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

pool2d.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

quantize.cu

CUDA: fix crash on large batch size for quant. MoE (#13537)

2025-05-14 16:41:02 +02:00

quantize.cuh

CUDA: batched+noncont MMQ, refactor bs>1 MoE code (#13199)

2025-04-30 23:12:59 +02:00

rope.cu

CUDA: backwards pass for misc. ops, add tests (#11257)

2025-01-16 16:43:38 +01:00

rope.cuh

RoPE: fix back, CUDA support for back + noncont. (#11240)

2025-01-15 12:51:37 +01:00

scale.cu

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

scale.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

softmax.cu

HIP: fix flash_attn_stream_k_fixup warning (#11604)

2025-02-02 23:48:29 +01:00

softmax.cuh

CUDA: backwards pass for misc. ops, add tests (#11257)

2025-01-16 16:43:38 +01:00

ssm-conv.cu

fix MUSA compiler warning (#12704)

2025-04-03 09:32:55 +02:00

ssm-conv.cuh

ggml : faster ssm scan (#10558)

2025-03-31 18:05:13 +02:00

ssm-scan.cu

CUDA/HIP: fix ssm_scan on devices where warp size is not 32 (#14196)

2025-06-15 17:30:13 +02:00

ssm-scan.cuh

ggml : faster ssm scan (#10558)

2025-03-31 18:05:13 +02:00

sum.cu

llama/ggml: add LLM training support (#10544)

2025-05-12 14:44:49 +02:00

sum.cuh

tests: add gradient tests for all backends (ggml/932)

2024-09-08 11:05:55 +03:00

sumrows.cu

CUDA: add mean operation (#14313)

2025-06-22 12:39:54 +08:00

sumrows.cuh

CUDA: add mean operation (#14313)

2025-06-22 12:39:54 +08:00

tsembd.cu

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

tsembd.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

unary.cu

ggml : add ggml_gelu_erf() CUDA kernel (#13719)

2025-05-24 13:06:47 +02:00

unary.cuh

ggml : add ggml_gelu_erf() CUDA kernel (#13719)

2025-05-24 13:06:47 +02:00

upscale.cu

musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)

2025-03-30 10:59:38 +02:00

upscale.cuh

llama : reorganize source code + improve CMake (#8006)

2024-06-26 18:33:02 +03:00

vecdotq.cuh

CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (#13014)

2025-04-22 21:27:40 +02:00

wkv.cu

llama: Add support for RWKV v7 architecture (#12412)

2025-03-18 07:27:50 +08:00

wkv.cuh

llama: Add support for RWKV v7 architecture (#12412)

2025-03-18 07:27:50 +08:00