From cbb901ac51bd6c41e4243ffb936ef0e2f7ca8ada Mon Sep 17 00:00:00 2001
From: Tim Dettmers <tim.dettmers@gmail.com>
Date: Tue, 26 Jul 2022 12:12:38 -0700
Subject: Boilerplate and test for extract_outliers.

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

(limited to 'csrc/kernels.cuh')

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