]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
feat: ref. cross entropy, add CUDA, fix grad test (#929)
authorJohannes Gäßler <redacted>
Tue, 27 Aug 2024 18:39:30 +0000 (20:39 +0200)
committerGitHub <redacted>
Tue, 27 Aug 2024 18:39:30 +0000 (20:39 +0200)
commit06fe976764b5cce6c4f0d25997af4fdf88a2d813
treec1aafa615b6f87818f12f948d92ae347d138c0aa
parent2438d62cb9290b5b5dc6228dec76fe81cf64238e
feat: ref. cross entropy, add CUDA, fix grad test (#929)
include/ggml-backend.h
src/ggml-cuda.cu
src/ggml-cuda/cross-entropy-loss.cu [new file with mode: 0644]
src/ggml-cuda/cross-entropy-loss.cuh [new file with mode: 0644]
src/ggml-cuda/sumrows.cu
src/ggml-cuda/sumrows.cuh
src/ggml.c
tests/test-backend-ops.cpp
tests/test-grad0.cpp