diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-16 10:56:17 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-16 10:56:17 -0700 |
| commit | 1ed2fa2f218d8dac401f3315420ffec92014c124 (patch) | |
| tree | 57863d4d1024689100c1b43caccc1d8739c58d99 /csrc/pythonInterface.c | |
| parent | 26efb154c8d77b4ede2cfc0dbd2381dd385f33e7 (diff) | |
Removed storage() from get_ptr; added boilerplate for bias dequant_mm.
Diffstat (limited to 'csrc/pythonInterface.c')
| -rw-r--r-- | csrc/pythonInterface.c | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/csrc/pythonInterface.c b/csrc/pythonInterface.c index 7356c11..0707674 100644 --- a/csrc/pythonInterface.c +++ b/csrc/pythonInterface.c @@ -248,8 +248,8 @@ extern "C" MAKE_FUNC_CTRANSFORM(8, col32, row, n, int8_t, COL32, ROW, false, 8) MAKE_FUNC_CTRANSFORM(32, col32, row, n, int32_t, COL32, ROW, false, 32) - void cdequant_mm_int32_fp16(int *A, float *rowStats, float *colStats, half *out, float* newRowStats, float* newcolStats, int numRows, int numCols) - { dequant_mm_int32_fp16(A, rowStats, colStats, out, newRowStats, newcolStats, numRows, numCols); } + void cdequant_mm_int32_fp16(int *A, float *rowStats, float *colStats, half *out, float* newRowStats, float* newcolStats, half* bias, int numRows, int numCols) + { dequant_mm_int32_fp16(A, rowStats, colStats, out, newRowStats, newcolStats, bias, numRows, numCols); } void cget_col_row_stats(half * A, float *rowStats, float *colStats, int *nnz_count_row, float nnz_threshold, int rows, int cols) { getColRowStats(A, rowStats, colStats, nnz_count_row, nnz_threshold, rows, cols); } |