From 8258b4364a21a4da2572cb644d0926080c3268da Mon Sep 17 00:00:00 2001
From: Max Ryabinin <mryabinin0@gmail.com>
Date: Fri, 1 Jul 2022 17:16:10 +0300
Subject: Add a CPU-only build option

---
 csrc/cpu_ops.h | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 csrc/cpu_ops.h

(limited to 'csrc/cpu_ops.h')

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
-- 
cgit v1.2.3