Aman Gupta 7de5c7cab6 CUDA: add set rows for f32 and f16 (#14551) ...

* CUDA: add set rows for f32 and f16

* Review: change kernel params, use strides from host

* Use 1-d kernel

* Review: use int64_t for blockDim.x, rename nb->s for clarity

2025-07-12 16:31:38 +03:00

..

template-instances

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

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

vendors

HIP : Add HIP 7.0+ compatibility for hipBLAS compute types (#14634)

2025-07-11 18:55:00 +02:00

acc.cu

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

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

acc.cuh

…

arange.cu

…

arange.cuh

…

argmax.cu

…

argmax.cuh

…

argsort.cu

…

argsort.cuh

…

binbcast.cu

…

binbcast.cuh

…

clamp.cu

…

clamp.cuh

…

CMakeLists.txt

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

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

common.cuh

musa: fix build warnings (unused variable) (#14561)

2025-07-08 07:58:30 +08:00

concat.cu

…

concat.cuh

…

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

…

conv-transpose-1d.cuh

…

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

…

count-equal.cuh

…

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

CUDA: add dynamic shared mem to softmax, refactor general usage (#14497)

2025-07-03 07:45:11 +08:00

cross-entropy-loss.cuh

…

dequantize.cuh

…

diagmask.cu

…

diagmask.cuh

…

fattn-common.cuh

CUDA: broadcasting for FlashAttention mask (#14500)

2025-07-02 15:48:33 +03:00

fattn-mma-f16.cuh

CUDA: broadcasting for FlashAttention mask (#14500)

2025-07-02 15:48:33 +03:00

fattn-tile-f16.cu

CUDA: broadcasting for FlashAttention mask (#14500)

2025-07-02 15:48:33 +03:00

fattn-tile-f16.cuh

…

fattn-tile-f32.cu

musa: fix build warnings (unused variable) (#14561)

2025-07-08 07:58:30 +08:00

fattn-tile-f32.cuh

…

fattn-vec-f16.cuh

CUDA: broadcasting for FlashAttention mask (#14500)

2025-07-02 15:48:33 +03:00

fattn-vec-f32.cuh

musa: fix build warnings (unused variable) (#14561)

2025-07-08 07:58:30 +08:00

fattn-wmma-f16.cu

CUDA: broadcasting for FlashAttention mask (#14500)

2025-07-02 15:48:33 +03:00

fattn-wmma-f16.cuh

…

fattn.cu

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

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

fattn.cuh

…

getrows.cu

CUDA: add bf16 and i32 to getrows (#14529)

2025-07-07 21:45:43 +08: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 set rows for f32 and f16 (#14551)

2025-07-12 16:31:38 +03:00

gla.cu

…

gla.cuh

…

im2col.cu

…

im2col.cuh

…

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

…

mmq.cu

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

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

mmq.cuh

CUDA: add dynamic shared mem to softmax, refactor general usage (#14497)

2025-07-03 07:45:11 +08: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

…

norm.cuh

…

opt-step-adamw.cu

…

opt-step-adamw.cuh

…

out-prod.cu

…

out-prod.cuh

…

pad.cu

…

pad.cuh

…

pool2d.cu

…

pool2d.cuh

…

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 : fix rope with partial rotation and non-cont src (#14580)

2025-07-08 10:15:21 +03:00

rope.cuh

…

scale.cu

ggml : add ggml_scale_bias (#14417)

2025-07-09 18:16:12 +02:00

scale.cuh

…

set-rows.cu

CUDA: add set rows for f32 and f16 (#14551)

2025-07-12 16:31:38 +03:00

set-rows.cuh

CUDA: add set rows for f32 and f16 (#14551)

2025-07-12 16:31:38 +03:00

softmax.cu

CUDA: add dynamic shared mem to softmax, refactor general usage (#14497)

2025-07-03 07:45:11 +08:00

softmax.cuh

…

ssm-conv.cu

model : support LiquidAI LFM2 hybrid family (#14620)

2025-07-11 20:27:01 +02:00

ssm-conv.cuh

…

ssm-scan.cu

cuda : support Falcon-H1 state size for SSM_SCAN (#14602)

2025-07-09 23:54:38 -04:00

ssm-scan.cuh

…

sum.cu

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

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

sum.cuh

…

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

…

tsembd.cuh

…

unary.cu

ggml : implement GEGLU_ERF and GEGLU_QUICK ops (#14445)

2025-07-03 23:07:22 +02:00

unary.cuh

ggml : implement GEGLU_ERF and GEGLU_QUICK ops (#14445)

2025-07-03 23:07:22 +02:00

upscale.cu

CUDA: add bilinear interpolation for upscale (#14563)

2025-07-08 10:11:18 +08:00

upscale.cuh

…

vecdotq.cuh

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

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

wkv.cu

…

wkv.cuh

…