diff options
| author | Determinant <[email protected]> | 2015-05-25 17:27:29 +0800 |
|---|---|---|
| committer | Determinant <[email protected]> | 2015-05-25 17:27:29 +0800 |
| commit | 81c115e09f9449ae61c7352edd77c68e9029f2dc (patch) | |
| tree | 284196680a794d4317d667e6a6e04fa5b3ee8c75 /matrix/cumatrix.c | |
| parent | c1e714052f0c2654ebe2a92d6961382146450b9e (diff) | |
add "copy from/to cumatrix" functionality
Diffstat (limited to 'matrix/cumatrix.c')
| -rw-r--r-- | matrix/cumatrix.c | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/matrix/cumatrix.c b/matrix/cumatrix.c index db4c784..51a3681 100644 --- a/matrix/cumatrix.c +++ b/matrix/cumatrix.c @@ -5,6 +5,7 @@ #define nerv_matrix_(NAME) nerv_matrix_cuda_float_##NAME #define cudak_(NAME) cudak_float_ ## NAME #define NERV_CUBLAS_(NAME) cublasS##NAME +#define MATRIX_CUMATRIX_HOST_TNAME nerv_matrix_host_float_tname const char *nerv_matrix_(tname) = "nerv.CuMatrixFloat"; #include "generic/cumatrix.c" #undef NERV_CUBLAS_ @@ -15,11 +16,13 @@ const char *nerv_matrix_(tname) = "nerv.CuMatrixFloat"; #undef MATRIX_ELEM #undef MATRIX_ELEM_PTR #undef MATRIX_ELEM_FMT +#undef MATRIX_CUMATRIX_HOST_TNAME #define MATRIX_USE_DOUBLE #define cuda_matrix_(NAME) cuda_matrix_double_##NAME #define nerv_matrix_(NAME) nerv_matrix_cuda_double_##NAME #define cudak_(NAME) cudak_double_ ## NAME #define NERV_CUBLAS_(NAME) cublasD##NAME +#define MATRIX_CUMATRIX_HOST_TNAME nerv_matrix_host_double_tname const char *nerv_matrix_(tname) = "nerv.CuMatrixDouble"; #include "generic/cumatrix.c" |