diff options
| author | Yimmon Zhuang <[email protected]> | 2015-09-14 16:14:49 +0800 |
|---|---|---|
| committer | Yimmon Zhuang <[email protected]> | 2015-09-14 16:14:49 +0800 |
| commit | 5b99c28961ca223cc35e77a4482eb789d5bef06d (patch) | |
| tree | eefc73df5c2dc535155d137f544f64ad0409bb6c /nerv/lib/matrix/cukernel.h | |
| parent | 667d3ab6a876feac022e889c260dd7abca5199e3 (diff) | |
| parent | 3721c74d56ffdea43851489617f33cd13b87ab76 (diff) | |
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'nerv/lib/matrix/cukernel.h')
| -rw-r--r-- | nerv/lib/matrix/cukernel.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/lib/matrix/cukernel.h b/nerv/lib/matrix/cukernel.h index 7bb4c2c..2126c6f 100644 --- a/nerv/lib/matrix/cukernel.h +++ b/nerv/lib/matrix/cukernel.h @@ -18,4 +18,6 @@ void cudak_(cuda_rearrange_frm)(const Matrix *a, Matrix *b, int step); void cudak_(cuda_scale_rows_by_row)(const Matrix *a, Matrix *b); void cudak_(cuda_scale_rows_by_col)(const Matrix *a, Matrix *b); void cudak_(cuda_decompress)(const Matrix *a, Matrix *b); +void cudak_(cuda_copy_rows_by_idx)(const Matrix *a, Matrix *b, + const Matrix *idx, int b_begin); #endif |