diff options
| author | txh18 <cloudygooseg@gmail.com> | 2015-12-02 15:23:54 +0800 |
|---|---|---|
| committer | txh18 <cloudygooseg@gmail.com> | 2015-12-02 15:23:54 +0800 |
| commit | 41a841f3e0992a578cf5c8f82ae44a552a6f8b2f (patch) | |
| tree | 0a5ba8d1530290fd91febcfe69986c96be0ac895 /nerv/lib/matrix/generic/cumatrix.c | |
| parent | 0c286aa6237da9e8daa7db8ed1e3805a33312926 (diff) | |
changed thres_mask function of matrix to a more standard api
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.c')
| -rw-r--r-- | nerv/lib/matrix/generic/cumatrix.c | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.c b/nerv/lib/matrix/generic/cumatrix.c index cbb0481..65e0788 100644 --- a/nerv/lib/matrix/generic/cumatrix.c +++ b/nerv/lib/matrix/generic/cumatrix.c @@ -83,9 +83,10 @@ void nerv_matrix_(rand_uniform)(Matrix *a, Status *status) { NERV_SET_STATUS(status, NERV_NORMAL, 0); } -void nerv_matrix_(thres_mask)(Matrix *a, double thres, double low, double high, Status *status) { +void nerv_matrix_(thres_mask)(Matrix *a, Matrix *b, double thres, double low, double high, Status *status) { + CHECK_SAME_DIMENSION(a, b, status); PROFILE_START - cudak_(cuda_thres_mask)(a, thres, low, high); + cudak_(cuda_thres_mask)(a, b, thres, low, high); PROFILE_STOP NERV_SET_STATUS(status, NERV_NORMAL, 0); } |