]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
overview / pkg / ggml / sources / llama.cpp / commit
summary | shortlog | log | commit | commitdiff | tree
(parent: 667d70d)
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>
Tue, 3 Dec 2024 18:04:49 +0000 (20:04 +0200)
commitefb6ae963031709fc331e6e48cc4606ac8f9c3a7
tree75af89915f68a251c88da298866a5dc920447b0btree
parent667d70d1704dfa6977505f5d01d4638669b90dcecommit | 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
tests/test-backend-ops.cpp diff | blob | history
Packaging of ggml-org/llama.cpp