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/cpu_ops.h | |
| parent | e2a75769f22bdc5465240c3f6701a1b002e8ab59 (diff) | |
| parent | 9b5f2eda8fbd3f042c4af7ed1b870525d4668f2a (diff) | |
Merge branch 'TimDettmers:main' into memory-efficient-backward
Diffstat (limited to 'csrc/cpu_ops.h')
| -rw-r--r-- | csrc/cpu_ops.h | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/csrc/cpu_ops.h b/csrc/cpu_ops.h index 57145a9..2ddf81e 100644 --- a/csrc/cpu_ops.h +++ b/csrc/cpu_ops.h @@ -1,9 +1,10 @@ #ifndef BITSANDBYTES_CPU_OPS_H #define BITSANDBYTES_CPU_OPS_H +#include <iostream> +#include <stdio.h> -void quantize_cpu(float *code, float *A, float *absmax, unsigned char *out, int n); - -void dequantize_cpu(float *code, unsigned char *A, float *absmax, float *out, int n); +void quantize_cpu(float *code, float *A, float *absmax, unsigned char *out, long long blocksize, long long n); +void dequantize_cpu(float *code, unsigned char *A, float *absmax, float *out, long long blocksize, long long n); #endif |