diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-26 12:12:38 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-26 12:12:38 -0700 |
| commit | cbb901ac51bd6c41e4243ffb936ef0e2f7ca8ada (patch) | |
| tree | f02615b5588aa6ed94a51c1e66297595b802c0a1 /csrc/ops.cuh | |
| parent | c771b3a75a6ebbfbfc398a028a477246b0799cf0 (diff) | |
Boilerplate and test for extract_outliers.
Diffstat (limited to 'csrc/ops.cuh')
| -rw-r--r-- | csrc/ops.cuh | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/csrc/ops.cuh b/csrc/ops.cuh index 4e719df..4b09ecf 100644 --- a/csrc/ops.cuh +++ b/csrc/ops.cuh @@ -174,4 +174,6 @@ void spmm_coo(cusparseHandle_t handle, int *A_rowidx, int *A_colidx, half *A_val template <typename T, int BITS> void spmm_coo_very_sparse_naive(int *max_count, int *max_idx, int *offset_rowidx, int *rowidx, int *colidx, half *values, T *B, half *out, float *dequant_stats, int nnz_rows, int nnz, int rowsA, int rowsB, int colsB); +template <int FORMAT> void extractOutliers(char * A, int *idx, char *out, int idx_size, int rows, int cols); + #endif |