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

about summary refs log tree commit diff

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

blob: 8467da79e0c71eb98e2525ef48b46c31afe83eb6 (plain)

#include "common.cuh"

#define CUDA_COUNT_EQUAL_CHUNK_SIZE 128

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