diff options
author | justheuristic <justheuristic@gmail.com> | 2022-09-17 18:42:22 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-09-17 18:42:22 +0300 |
commit | 3634fc738bc20e4041c75544d3f678f61ce2348c (patch) | |
tree | 36bc3394748ce4141fa9ab9d1104ca6441ade65c /csrc/pythonInterface.c | |
parent | e2a75769f22bdc5465240c3f6701a1b002e8ab59 (diff) | |
parent | 9b5f2eda8fbd3f042c4af7ed1b870525d4668f2a (diff) |
Merge branch 'TimDettmers:main' into memory-efficient-backward
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 0707674..58e26a9 100644 --- a/csrc/pythonInterface.c +++ b/csrc/pythonInterface.c @@ -287,7 +287,7 @@ extern "C" void cextractOutliers_ampere(char * A, int *idx, char *out, int idx_size, int rows, int cols){ extractOutliers_ampere(A, idx, out, idx_size, rows, cols); } #endif - void cquantize_blockwise_cpu_fp32(float *code, float *A, float *absmax, unsigned char *out, const int n){ quantize_cpu(code, A, absmax, out, n); } - void cdequantize_blockwise_cpu_fp32(float *code, unsigned char *A, float *absmax, float *out, const int n){ dequantize_cpu(code, A, absmax, out, n); } + void cquantize_blockwise_cpu_fp32(float *code, float *A, float *absmax, unsigned char *out, long long blocksize, long long n){ quantize_cpu(code, A, absmax, out, blocksize, n); } + void cdequantize_blockwise_cpu_fp32(float *code, unsigned char *A, float *absmax, float *out, long long blocksize, long long n){ dequantize_cpu(code, A, absmax, out, blocksize, n); } } |