diff options
| author | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:16:10 +0300 |
|---|---|---|
| committer | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:16:10 +0300 |
| commit | 8258b4364a21a4da2572cb644d0926080c3268da (patch) | |
| tree | 571e95bc327116fbaba08d14871fb0b224b8a65b /csrc/cpu_ops.h | |
| parent | 33efe4a09f459832e8beceba70add0695cc485e4 (diff) | |
Add a CPU-only build option
Diffstat (limited to 'csrc/cpu_ops.h')
| -rw-r--r-- | csrc/cpu_ops.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/csrc/cpu_ops.h b/csrc/cpu_ops.h new file mode 100644 index 0000000..57145a9 --- /dev/null +++ b/csrc/cpu_ops.h @@ -0,0 +1,9 @@ +#ifndef BITSANDBYTES_CPU_OPS_H +#define BITSANDBYTES_CPU_OPS_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); + +#endif |