]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
feat: add `GGML_UNARY_OP_ARGMAX` Metal kernel (#1019)
authorPAB <redacted>
Mon, 2 Dec 2024 18:27:24 +0000 (19:27 +0100)
committerGitHub <redacted>
Mon, 2 Dec 2024 18:27:24 +0000 (19:27 +0100)
commit589fed13a77d7e54435c2182384955706b60b841
treeeaab59091165ca9b797f92462d9fdf736ed89338
parentd8ea053461056a5c15f071c7c5ed57d86e892750
feat: add `GGML_UNARY_OP_ARGMAX` Metal kernel (#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>
src/ggml-metal/ggml-metal.m
src/ggml-metal/ggml-metal.metal
tests/test-backend-ops.cpp