char-fork/llama.cpp

git clone https://git.t4t.associates/char-fork/llama.cpp

Piotr Wilkin (ilintar)Add DIAG for CUDA (#17873)b63509262

char/aliceai-t5-moe
133 B5 linesraw
1#include "common.cuh"
2
3#define CUDA_DIAG_BLOCK_SIZE 256
4
5void ggml_cuda_op_diag(ggml_backend_cuda_context & ctx, ggml_tensor * dst);