Arrcttacsrks commited on
Commit
2461a46
·
verified ·
1 Parent(s): deb4179

Upload llama.cpp/ggml/src/ggml-cuda/diagmask.cuh with huggingface_hub

Browse files
llama.cpp/ggml/src/ggml-cuda/diagmask.cuh ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ #include "common.cuh"
2
+
3
+ #define CUDA_DIAG_MASK_INF_BLOCK_SIZE 32
4
+
5
+ void ggml_cuda_op_diag_mask_inf(ggml_backend_cuda_context & ctx, ggml_tensor * dst);