Genericize index type

1 files changed, 8 insertions, 8 deletions

diff --git a/ssse3_gemm.h b/ssse3_gemm.h
index 07bfb5e..4993ef6 100644
--- a/ssse3_gemm.h
+++ b/ssse3_gemm.h
@@ -1,5 +1,5 @@
 #pragma once
-#include "cpu_type.h"
+#include "types.h"
 #include <cstdint>
 #include <stdint.h>
 
@@ -12,21 +12,21 @@ struct SSSE3_8bit {
   typedef int8_t Integer;
 
   // Currently A is prepared by quantization but this could theoretically change.
-  static inline void PrepareA(const float *input, int8_t *output, float quant_mult, int rows, int cols) {
+  static inline void PrepareA(const float *input, int8_t *output, float quant_mult, Index rows, Index cols) {
     Quantize(input, output, quant_mult, rows * cols);
   }
 
-  static void Quantize(const float *input, int8_t *output, float quant_mult, int size);
+  static void Quantize(const float *input, int8_t *output, float quant_mult, Index size);
 
   // Tile size for B; B must be a multiple of this block size.
-  static const int kBTileRow = 16;
-  static const int kBTileCol = 8;
+  static const Index kBTileRow = 16;
+  static const Index kBTileCol = 8;
 
-  static void PrepareB(const float *input, int8_t *output, float quant_mult, int rows, int cols);
+  static void PrepareB(const float *input, int8_t *output, float quant_mult, Index rows, Index cols);
 
-  static void SelectColumnsB(const int8_t *input, int8_t *output, int rows, const std::size_t *cols_begin, const std::size_t *cols_end);
+  static void SelectColumnsB(const int8_t *input, int8_t *output, Index rows, const Index *cols_begin, const Index *cols_end);
 
-  static void Multiply(const int8_t *A, const int8_t *B, float *C, float unquant_mult, int A_rows, int width, int B_cols);
+  static void Multiply(const int8_t *A, const int8_t *B, float *C, float unquant_mult, Index A_rows, Index width, Index B_cols);
   
   static const char *const kName;