PR #10441 cuda : optimize argmax

Commits

cuda : optimize argmax

slaren committed 1 year ago
remove unused parameter

slaren committed 1 year ago
fixup : use full warps

slaren committed 1 year ago
Apply suggestions from code review

slaren committed 1 year ago
fix ub

slaren committed 1 year ago
ggml : check ne00 <= INT32_MAX in argmax and argsort

slaren committed 1 year ago

llama.cpp cuda : optimize argmax #10441 Merged