argmax.cuh « ggml-cuda « src « ggml « llama.cpp - llmnpc - Testing how LLM's fair as NPC's

summary refs log tree commit diff

path: root/llama.cpp/ggml/src/ggml-cuda/argmax.cuh

blob: 5b7223adc6baa58bc536ab59ff60e9d98f996465 (plain)

1
2
3

#include "common.cuh"

void ggml_cuda_argmax(ggml_backend_cuda_context & ctx, ggml_tensor * dst);