aboutsummaryrefslogtreecommitdiff
path: root/nerv/lib/matrix/cumatrix.h
diff options
context:
space:
mode:
Diffstat (limited to 'nerv/lib/matrix/cumatrix.h')
-rw-r--r--nerv/lib/matrix/cumatrix.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/lib/matrix/cumatrix.h b/nerv/lib/matrix/cumatrix.h
index e6def66..b47e14b 100644
--- a/nerv/lib/matrix/cumatrix.h
+++ b/nerv/lib/matrix/cumatrix.h
@@ -1,7 +1,9 @@
#ifndef NERV_CUMATRIX_H
#define NERV_CUMATRIX_H
#include "matrix.h"
+#include "../common.h"
void nerv_cumatrix_print_profile();
void nerv_cumatrix_clear_profile();
void nerv_cumatrix_init();
+void nerv_cumatrix_select_gpu(int dev, Status *status);
#endif