]> git.djapps.eu Git - pkg/ggml/sources/ggml/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 08:52:14 +0000 (11:52 +0300)
commit2a7883af5d456b32384000499c8a1e68da4acf91
treef6769ff33096b5342d473f6a72234d86f82bf405
parentb05568b3cb7277534bb840bffa78ab4b8676adb7
CUDA: add mean operation (llama/14313)

* CUDA: add mean operation

* add back sum_rows_f32_cuda

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