From bcab99ec877ba063543bd7c03ba1cdd1b06e8078 Mon Sep 17 00:00:00 2001
From: Tim Dettmers <tim.dettmers@gmail.com>
Date: Tue, 26 Jul 2022 17:39:30 -0700
Subject: Working outlier extraction for Turing.

---
 csrc/kernels.cuh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'csrc/kernels.cuh')

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
 
-- 
cgit v1.2.3