diff options
Diffstat (limited to 'nerv/lib/matrix/matrix.h')
| -rw-r--r-- | nerv/lib/matrix/matrix.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/nerv/lib/matrix/matrix.h b/nerv/lib/matrix/matrix.h index a28fd97..35b698f 100644 --- a/nerv/lib/matrix/matrix.h +++ b/nerv/lib/matrix/matrix.h @@ -2,7 +2,6 @@ #define NERV_GENERIC_MATRIX_H #include <stddef.h> -#include <curand.h> typedef struct Matrix { size_t stride; /* size of a row */ @@ -15,9 +14,10 @@ typedef struct Matrix { } data; /* pointer to actual storage */ unsigned long offset; /* the actual beginning of the matrix */ long *data_ref; - curandGenerator_t *curand_gen; } Matrix; #define MATRIX_ROW_PTR(self, row) \ (MATRIX_ELEM *)((char *)MATRIX_ELEM_PTR(self) + (row) * (self)->stride) +#define MATRIX_NEXT_ROW_PTR(self, stride) \ + (MATRIX_ELEM *)((char *)self + stride) #endif |