diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-26 12:12:38 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-26 12:12:38 -0700 |
| commit | cbb901ac51bd6c41e4243ffb936ef0e2f7ca8ada (patch) | |
| tree | f02615b5588aa6ed94a51c1e66297595b802c0a1 /csrc/kernels.cuh | |
| parent | c771b3a75a6ebbfbfc398a028a477246b0799cf0 (diff) | |
Boilerplate and test for extract_outliers.
Diffstat (limited to 'csrc/kernels.cuh')
| -rw-r--r-- | csrc/kernels.cuh | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/csrc/kernels.cuh b/csrc/kernels.cuh index cbfbeba..ec2068e 100644 --- a/csrc/kernels.cuh +++ b/csrc/kernels.cuh @@ -118,6 +118,8 @@ template <int THREADS, int ITEMS_PER_THREAD, int TILE_ROWS, int TILE_COLS, int S template <int THREADS, int ITEMS_PER_THREAD, int TILE_ROWS, int TILE_COLS, int TRANSPOSE, int FORMAT> __global__ void kTransformRowToFormat(char *__restrict__ const A, char *out, int rows, int cols, int tiledCols, int outRows, int outCols); +template <int FORMAT> __global__ void kExtractOutliers(char *A, int *idx, char *out, int rowsA, int colsA, int tiledRowsA, int tiledColsA); + #endif |