]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commit
feat: ref. cross entropy, add CUDA, fix grad test (ggml/929)
authorJohannes Gäßler <redacted>
Tue, 27 Aug 2024 18:39:30 +0000 (20:39 +0200)
committerGeorgi Gerganov <redacted>
Wed, 28 Aug 2024 10:22:20 +0000 (13:22 +0300)
commit8954769aa2b25059fef2dfd235a4aded128c7824
tree15205a3fd5fbbda3e10b535c419b7b20b4414fa3
parentdf06468d9e61cfb99651ebd9b843b1b1524449e4
feat: ref. cross entropy, add CUDA, fix grad test (ggml/929)
ggml/include/ggml-backend.h
ggml/src/ggml-cuda.cu
ggml/src/ggml-cuda/cross-entropy-loss.cu [new file with mode: 0644]
ggml/src/ggml-cuda/cross-entropy-loss.cuh [new file with mode: 0644]
ggml/src/ggml-cuda/sumrows.cu
ggml/src/ggml-cuda/sumrows.cuh
ggml/src/ggml.c