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/ops.cu | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

(limited to 'csrc/ops.cu')

diff --git a/csrc/ops.cu b/csrc/ops.cu
index fe2d7fe..e6227ae 100644
--- a/csrc/ops.cu
+++ b/csrc/ops.cu
@@ -586,8 +586,7 @@ template <int FORMAT> void extractOutliers(char * A, int *idx, char *out, int id
   int tiledCols = tiledCols = fill_up_to_nearest_multiple(cols, 32);
   int tiledRows = 0;
 
-	int elements = idx_size*cols; // matrix A is transposed, so we extract columns
-	int num_blocks = (elements+threads-1)/threads;
+	int num_blocks = idx_size;
 
   if(FORMAT == COL_TURING)
   {
@@ -598,7 +597,7 @@ template <int FORMAT> void extractOutliers(char * A, int *idx, char *out, int id
       tiledRows = fill_up_to_nearest_multiple(rows, 32);
 	}
 
-  kExtractOutliers<FORMAT><<<num_blocks, threads>>>(A, idx, out, rows, cols, tiledRows, tiledCols);
+  kExtractOutliers<FORMAT><<<num_blocks, threads>>>(A, idx, out, idx_size, rows, cols, tiledRows, tiledCols);
   CUDA_CHECK_RETURN(cudaPeekAtLastError());
 }
 
-- 
cgit v1.2.3