cumsum.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/cumsum.cuh

blob: 782d1d92e9bb1d9589df9e30c4d8f1851c179532 (plain)

#include "common.cuh"

#define CUDA_CUMSUM_BLOCK_SIZE 256

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