Set up cmake to compile library

2025-11-07 10:14:27 +00:00 · 2024-02-17 23:07:09 +01:00
parent ac18768297
commit f541e2f7f8
7 changed files with 40 additions and 93 deletions
--- a/src/layers/dense.cpp
+++ b/src/layers/dense.cpp
@@ -1,6 +1,7 @@
 #include "dense.h"
 #include "cuda_helper.h"
 #include <cstdlib>
+#include <cuda_runtime.h>
 #include <cublas_v2.h>
 #include <cstdio>
 #include <stdexcept>