]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commit
overview / pkg / ggml / sources / whisper.cpp / commit
summary | shortlog | log | commit | commitdiff | tree
(parent: 5773a14)
feat: add `GGML_UNARY_OP_ARGMAX` Metal kernel (ggml/1019)
authorPAB <redacted>
Mon, 2 Dec 2024 18:27:24 +0000 (19:27 +0100)
committerGeorgi Gerganov <redacted>
Sun, 8 Dec 2024 18:14:35 +0000 (20:14 +0200)
commitc52d1035de6ddaa388f2a5d4bfd0713d447a4ae8
treecbafcd2896fb875dbceec94d2159b893dd124e34tree
parent5773a14980f62a95cb20b1499bf207c76e9a8ce1commit | diff
feat: add `GGML_UNARY_OP_ARGMAX` Metal kernel (ggml/1019)

* implemented argmax kernel

* tpig -> tgpig

* change to strides

* contiguous assertions

* kernel working and tested

* argmax simd parallel implementation

* added 2 new tests for argmax in test-backend-ops

* cosmit

* added 3 tests cases for perf eval

* add test_argmax in make_test_cases_perf

* Update test-backend-ops.cpp

Co-authored-by: Diego Devesa <redacted>
---------

Co-authored-by: Diego Devesa <redacted>
ggml/src/ggml-metal/ggml-metal.m diff | blob | history
ggml/src/ggml-metal/ggml-metal.metal diff | blob | history
Packaging of ggerganov/whisper.cpp