xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/RuyUtils.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #pragma once
2 
3 #ifdef USE_RUY_QMATMUL
4 
5 #include <ruy/ruy.h>
6 
7 namespace at {
8 namespace native {
9 namespace ruy_utils {
10 
11 ruy::Context* get_ruy_context();
12 
13 void quantize_multiplier(double scale,
14                          int* multiplier_fixedpoint,
15                          int* multiplier_exponent);
16 
17 } // namespace ruy_utils
18 } // namespace native
19 } // namespace
20 
21 #endif // USE_RUY_QMATMUL
22