0 0 0 C++

cuda : optimize argmax (#10441)

* cuda : optimize argmax

* remove unused parameter

ggml-ci

* fixup : use full warps

ggml-ci

* Apply suggestions from code review

Co-authored-by: Johannes Gäßler <johannesg@5d6.de>

* fix ub

* ggml : check ne00 <= INT32_MAX in argmax and argsort

---------

Co-authored-by: Johannes Gäßler <johannesg@5d6.de>

Diego Devesa committed 1y ago

a5e47592b6171ae21f3eaa1aba6fb2b707875063

Parent: 1bb30bf

Committed by GitHub <noreply@github.com> on 11/21/2024, 5:18:50 PM