Johannes Gäßler 73e2ed3ce3 CUDA: use async data loading for FlashAttention (#11894) ...

* CUDA: use async data loading for FlashAttention

---------

Co-authored-by: Diego Devesa <slarengh@gmail.com>

2025-02-17 14:03:24 +01:00

..

template-instances

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

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

vendors

CUDA/HIP: add support for selectable warp size to mmv (#11519)

2025-02-02 22:40:09 +01:00

acc.cu

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

2024-06-26 18:33:02 +03: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

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

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

binbcast.cuh

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

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

clamp.cu

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

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

clamp.cuh

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

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

CMakeLists.txt

cuda : add ampere to the list of default architectures (#11870)

2025-02-14 15:33:52 +01:00

common.cuh

CUDA: use async data loading for FlashAttention (#11894)

2025-02-17 14:03:24 +01:00

concat.cu

fix: add missing msg in static_assert (#11143)

2025-01-08 20:03:28 +00:00

concat.cuh

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

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

conv-transpose-1d.cu

feat: cuda implementation for ggml_conv_transpose_1d (ggml/854)

2024-07-08 12:23:00 +03: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: use arch list for compatibility check (#11775)

2025-02-11 00:17:22 +01:00

convert.cuh

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

2024-06-26 18:33: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-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: use async data loading for FlashAttention (#11894)

2025-02-17 14:03:24 +01:00

cpy.cu

cuda: add q8_0->f32 cpy operation (#9571)

2024-09-24 02:14:24 +02:00

cpy.cuh

increase cuda_cpy block size (ggml/996)

2024-10-26 10:33:56 +03:00

cross-entropy-loss.cu

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

2025-01-16 16:43:38 +01: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: use async data loading for FlashAttention (#11894)

2025-02-17 14:03:24 +01:00

fattn-mma-f16.cuh

CUDA: use async data loading for FlashAttention (#11894)

2025-02-17 14:03:24 +01:00

fattn-tile-f16.cu

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

2025-02-02 19:31:09 +01: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: use mma PTX instructions for FlashAttention (#11583)

2025-02-02 19:31:09 +01: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: use mma PTX instructions for FlashAttention (#11583)

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

fattn-vec-f32.cuh

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

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

fattn-wmma-f16.cu

CUDA: fix Volta FlashAttention logic (#11615)

2025-02-03 14:25:56 +02:00

fattn-wmma-f16.cuh

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

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

fattn.cu

CUDA: fix Volta FlashAttention logic (#11615)

2025-02-03 14:25:56 +02:00

fattn.cuh

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

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

getrows.cu

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

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

getrows.cuh

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

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

ggml-cuda.cu

musa: bump MUSA SDK version to rc3.1.1 (#11822)

2025-02-13 13:28:18 +01: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

mma.cuh

CUDA: use async data loading for FlashAttention (#11894)

2025-02-17 14:03:24 +01:00

mmq.cu

HIP: Remove GCN from list of devices that avoid MMQ (#11831)

2025-02-12 22:25:28 +01:00

mmq.cuh

CUDA: use async data loading for FlashAttention (#11894)

2025-02-17 14:03:24 +01:00

mmv.cu

CUDA: support for mat. mul. with ne03 != ne13 (#11656)

2025-02-05 08:58:31 +01:00

mmv.cuh

CUDA: remove DMMV, consolidate F16 mult mat vec (#10318)

2024-11-17 09:09:55 +01:00

mmvq.cu

rocBLAS: Avoid fp32->fp16->fp32 conversion on cdna (#11356)

2025-01-24 17:50:49 +01:00

mmvq.cuh

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

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

norm.cu

CUDA: non-contiguous (RMS) norm support (#11659)

2025-02-04 22:21:42 +01:00

norm.cuh

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

2025-01-16 16:43:38 +01: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

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

2024-06-26 18:33:02 +03: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 : optimize argmax (#10441)

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

quantize.cuh

CUDA: optimize and refactor MMQ (#8416)

2024-07-11 16:47:47 +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

sum.cu

CUDA: fix CUDART_VERSION checks (#11821)

2025-02-12 13:16:39 +01:00

sum.cuh

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

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

sumrows.cu

sync : ggml

2024-08-27 22:41:27 +03:00

sumrows.cuh

sync : ggml

2024-08-27 22:41:27 +03: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

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

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

unary.cuh

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

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

upscale.cu

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

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

upscale.cuh

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

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

vecdotq.cuh

CUDA: MMQ code deduplication + iquant support (#8495)

2024-07-20 22:25:26 +02:00

wkv6.cu

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

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

wkv6.cuh

Optimize RWKV6 Operator Naming and Implement Multi-core CPU/ SYCL Acceleration (#10133)

2024-11-07 15:19:10 +08:00