]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
CUDA: add mean operation (#14313)
authorAman Gupta <redacted>
Sun, 22 Jun 2025 04:39:54 +0000 (12:39 +0800)
committerGitHub <redacted>
Sun, 22 Jun 2025 04:39:54 +0000 (12:39 +0800)
commitaa064b2eb7f7779db5e094a9d8d66d5033557de2
tree9453e3e8eca9384c5973f3f6b138c2b171d0fa0a
parentaa0ef5c578eef4c2adc7be1282f21bab5f3e8d26
CUDA: add mean operation (#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
tests/test-backend-ops.cpp