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

blob: 7d73e6a8eb44cd877b3e7a929fc41438ccbbeb85 (plain)

#include "common.cuh"

#define CUDA_DIAG_BLOCK_SIZE 256

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