1 files changed, 9 insertions, 0 deletions
diff --git a/test/TestUtils.h b/test/TestUtils.h
index 2cb7b88..d320ae2 100644
--- a/test/TestUtils.h
+++ b/test/TestUtils.h
@@ -7,9 +7,18 @@
 #pragma once
 #include <cmath>
 #include <vector>
+#include "fbgemm/Fbgemm.h"
 
 namespace fbgemm {
 
+static std::vector<matrix_op_t> transposeVals = { matrix_op_t::NoTranspose,
+                                    matrix_op_t::Transpose };
+
+static std::vector<QuantizationGranularity> qGranularityVals = {
+      QuantizationGranularity::TENSOR,
+      QuantizationGranularity::GROUP,
+      QuantizationGranularity::OUT_CHANNEL };
+
 /*
  * @brief Check and validate the buffers for reference and FBGEMM result.
  */