diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-04 07:40:48 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-04 07:40:48 -0700 |
| commit | cc5b323876392658b1d91655f30840d24be6d821 (patch) | |
| tree | 8e23e961709a3cc082a707ebc8ea0f52baee6923 /csrc/kernels.cuh | |
| parent | 6101a8fb9f76c2cc4018452b4420dd52e946d52b (diff) | |
| parent | bd515328d70f344f935075f359c5aefc616878d5 (diff) | |
Merge branch 'extract_outliers' into debug
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 4e65e96..2447494 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 idx_size, int rowsA, int colsA, int tiledRowsA, int tiledColsA); + #endif |