SIGN IN SIGN UP

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 <slarengh@gmail.com>

---------

Co-authored-by: Diego Devesa <slarengh@gmail.com>
P
PAB committed
efb6ae963031709fc331e6e48cc4606ac8f9c3a7
Parent: 667d70d
Committed by Georgi Gerganov <ggerganov@gmail.com> on 12/3/2024, 6:04:49 PM