Add DIAG for CUDA (#17873)

* Add DIAG for CUDA * Refactor parameters
2025-12-09 20:28:57 +01:00
parent 48f47565a7
commit b63509262a
4 changed files with 116 additions and 0 deletions
@@ -0,0 +1,5 @@
+#include "common.cuh"
+
+#define CUDA_DIAG_BLOCK_SIZE 256
+
+void ggml_cuda_op_diag(ggml_backend_cuda_context & ctx, ggml_tensor * dst);