diff options
author | Determinant <ted.sybil@gmail.com> | 2016-02-16 14:50:31 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-02-16 14:50:31 +0800 |
commit | 9e7171e2da3e4edba303f5c2bdaef416fb62e81a (patch) | |
tree | 722605d92482df5f57e329ab4b3e34b6982671ee /nerv/lib/matrix/generic/cumatrix.h | |
parent | a7608a67f16f745309409f6a004354121e0b3ba6 (diff) |
pass cflags to nvcc correctly; add omitted func decl
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.h')
-rw-r--r-- | nerv/lib/matrix/generic/cumatrix.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.h b/nerv/lib/matrix/generic/cumatrix.h index 9304060..4f66a2c 100644 --- a/nerv/lib/matrix/generic/cumatrix.h +++ b/nerv/lib/matrix/generic/cumatrix.h @@ -65,3 +65,10 @@ void nerv_matrix_(thres_mask)(Matrix *a, Matrix *b, double thres, double low, double high, Status *status); void nerv_matrix_(rand_uniform)(Matrix *a, Status *status); + +#ifdef __NERV_FUTURE_CUDA_7 +void nerv_matrix_(update_select_rows_by_rowidx)(Matrix *c, const Matrix *a, const Matrix *idx, + double alpha, double beta, Status *status); +void nerv_matrix_(update_select_rows_by_colidx)(Matrix *c, const Matrix *a, const Matrix *idx, + double alpha, double beta, Status *status); +#endif |