]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commit
CUDA: add mean operation (llama/14313)
authorAman Gupta <redacted>
Sun, 22 Jun 2025 04:39:54 +0000 (12:39 +0800)
committerGeorgi Gerganov <redacted>
Tue, 1 Jul 2025 14:54:53 +0000 (17:54 +0300)
commitabb650c0ec77a32a971189bb2d10bbb20e394993
treeb384a29b4fa46e7dacc503206aec87cde3a86d93
parente0366767952a928106f82c4dddaff58520b865ab
CUDA: add mean operation (llama/14313)

* CUDA: add mean operation

* add back sum_rows_f32_cuda

* Review: early exit if col!=0
ggml/src/ggml-cuda/common.cuh
ggml/src/ggml-cuda/ggml-cuda.cu
ggml/src/ggml-cuda/mean.cu [new file with mode: 0644]
ggml/src/ggml-cuda/mean.cuh [new file with mode: 0644]
ggml/src/ggml-cuda/sumrows.cu
ggml/src/ggml-cuda/sumrows.cuh