diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/fbgemm/Fbgemm.h | 4 | ||||
-rw-r--r-- | include/fbgemm/FbgemmI8Spmdm.h | 2 | ||||
-rw-r--r-- | include/fbgemm/OutputProcessing-inl.h | 2 | ||||
-rw-r--r-- | include/fbgemm/QuantUtils.h | 3 | ||||
-rw-r--r-- | include/fbgemm/Utils.h | 2 |
5 files changed, 6 insertions, 7 deletions
diff --git a/include/fbgemm/Fbgemm.h b/include/fbgemm/Fbgemm.h index 3cf832c..b729d56 100644 --- a/include/fbgemm/Fbgemm.h +++ b/include/fbgemm/Fbgemm.h @@ -15,11 +15,11 @@ #include <memory> #include <type_traits> #include "ConvUtils.h" +#include "FbgemmBuild.h" #include "FbgemmI8Spmdm.h" +#include "QuantUtilsAvx2.h" #include "Types.h" #include "Utils.h" -#include "FbgemmBuild.h" -#include "QuantUtilsAvx2.h" // Turning on this option will print out time breakdown of each stage (e.g., // input packing, the main GEMM kernel, each output processing pipeline). diff --git a/include/fbgemm/FbgemmI8Spmdm.h b/include/fbgemm/FbgemmI8Spmdm.h index 01416d5..93af6ea 100644 --- a/include/fbgemm/FbgemmI8Spmdm.h +++ b/include/fbgemm/FbgemmI8Spmdm.h @@ -9,8 +9,8 @@ #include <cstdint> #include <vector> #include "ConvUtils.h" -#include "Utils.h" #include "FbgemmBuild.h" +#include "Utils.h" // #define FBGEMM_MEASURE_TIME_BREAKDOWN diff --git a/include/fbgemm/OutputProcessing-inl.h b/include/fbgemm/OutputProcessing-inl.h index 2ff64f3..c250942 100644 --- a/include/fbgemm/OutputProcessing-inl.h +++ b/include/fbgemm/OutputProcessing-inl.h @@ -111,7 +111,7 @@ inline int ReQuantizeOutput<FUSE_RELU, Q_GRAN, outT, inT, nextOPType>::f( } else if (instSet == inst_set_t::avx2 || instSet == inst_set_t::avx512) { if (std::is_same<outT, uint8_t>::value) { bool b_symmetric = (Q_GRAN == QuantizationGranularity::TENSOR && - Bq_zero_point_[0] == 0) || + Bq_zero_point_[0] == 0) || q_row_offsets_ == nullptr; requantizationParams_t r = {Aq_zero_point_, diff --git a/include/fbgemm/QuantUtils.h b/include/fbgemm/QuantUtils.h index f658f73..43855d8 100644 --- a/include/fbgemm/QuantUtils.h +++ b/include/fbgemm/QuantUtils.h @@ -68,8 +68,7 @@ FBGEMM_API T Quantize( template <typename T> FBGEMM_API T Quantize(float src, const TensorQuantizationParams& qparams) { - return Quantize<T>( - src, qparams.zero_point, qparams.scale, qparams.precision); + return Quantize<T>(src, qparams.zero_point, qparams.scale, qparams.precision); } template <typename T> diff --git a/include/fbgemm/Utils.h b/include/fbgemm/Utils.h index a7e91cd..775e18c 100644 --- a/include/fbgemm/Utils.h +++ b/include/fbgemm/Utils.h @@ -7,8 +7,8 @@ #pragma once #include <string> #include <type_traits> -#include "UtilsAvx2.h" #include "FbgemmBuild.h" +#include "UtilsAvx2.h" namespace fbgemm { |