diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-26 17:39:30 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-26 17:39:30 -0700 |
| commit | bcab99ec877ba063543bd7c03ba1cdd1b06e8078 (patch) | |
| tree | 3d1dcfec5d3361ad4e5c5ba552a432e045445851 /csrc/kernels.cuh | |
| parent | cbb901ac51bd6c41e4243ffb936ef0e2f7ca8ada (diff) | |
Working outlier extraction for Turing.
Diffstat (limited to 'csrc/kernels.cuh')
| -rw-r--r-- | csrc/kernels.cuh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/csrc/kernels.cuh b/csrc/kernels.cuh index ec2068e..eda5ba0 100644 --- a/csrc/kernels.cuh +++ b/csrc/kernels.cuh @@ -118,7 +118,7 @@ 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); +template <int FORMAT> __global__ void kExtractOutliers(char *A, int *idx, char *out, int idx_size, int rowsA, int colsA, int tiledRowsA, int tiledColsA); #endif |