From 8258b4364a21a4da2572cb644d0926080c3268da Mon Sep 17 00:00:00 2001 From: Max Ryabinin Date: Fri, 1 Jul 2022 17:16:10 +0300 Subject: Add a CPU-only build option --- csrc/common.h | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 csrc/common.h (limited to 'csrc/common.h') diff --git a/csrc/common.h b/csrc/common.h new file mode 100644 index 0000000..35f2463 --- /dev/null +++ b/csrc/common.h @@ -0,0 +1,23 @@ +#include + +#ifndef common +#define common + +using namespace BinSearch; + +struct quantize_block_args { + BinAlgo *bin_searcher; + float *code; + float *A; + float *absmax; + unsigned char *out; + int block_end; + int block_idx; + int threadidx; +}; + +#define BLOCK_SIZE 4096 + +void *quantize_block(void *arguments); + +#endif \ No newline at end of file -- cgit v1.2.3 From 31ce1b3708751016bf5e14beff7ae0a99c975991 Mon Sep 17 00:00:00 2001 From: Max Ryabinin Date: Fri, 1 Jul 2022 17:36:30 +0300 Subject: Reduce diff --- csrc/common.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'csrc/common.h') diff --git a/csrc/common.h b/csrc/common.h index 35f2463..2f25a58 100644 --- a/csrc/common.h +++ b/csrc/common.h @@ -20,4 +20,4 @@ struct quantize_block_args { void *quantize_block(void *arguments); -#endif \ No newline at end of file +#endif -- cgit v1.2.3