diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-04 07:40:48 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-04 07:40:48 -0700 |
| commit | cc5b323876392658b1d91655f30840d24be6d821 (patch) | |
| tree | 8e23e961709a3cc082a707ebc8ea0f52baee6923 /csrc/ops.cuh | |
| parent | 6101a8fb9f76c2cc4018452b4420dd52e946d52b (diff) | |
| parent | bd515328d70f344f935075f359c5aefc616878d5 (diff) | |
Merge branch 'extract_outliers' into debug
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 |