Fixed 2^31 max size issue for cpu blockwise quant.

1 files changed, 6 insertions, 4 deletions

diff --git a/csrc/common.h b/csrc/common.h
index 2f25a58..c99034e 100644
--- a/csrc/common.h
+++ b/csrc/common.h
@@ -5,18 +5,20 @@
 
 using namespace BinSearch;
 
+#define BLOCK_SIZE 16384
+
 struct quantize_block_args {
     BinAlgo<Scalar, float, Direct2> *bin_searcher;
     float *code;
     float *A;
     float *absmax;
     unsigned char *out;
-    int block_end;
-    int block_idx;
-    int threadidx;
+    long long block_end;
+    long long block_idx;
+    long long threadidx;
+		long long blocksize;
 };
 
-#define BLOCK_SIZE 4096
 
 void *quantize_block(void *arguments);