Fix for int4 tuning (apache#37)

* set lower precision * add flop for conv2d on arm cpu
tmoreau89 · Mar 21, 2019 · cf39305 · cf39305
1 parent a98b713
commit cf39305
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/tvm/autotvm/measure/measure_methods.py b/python/tvm/autotvm/measure/measure_methods.py
@@ -481,7 +481,7 @@ def run_through_rpc(measure_input, build_result,
         # check correctness of output
         if ref_output:
             for expected, real in zip(ref_output, args):
-                if not np.allclose(expected, real.asnumpy(), rtol=1e-4):
+                if not np.allclose(expected, real.asnumpy(), rtol=1e-2):
                     logger.warning("Wrong Answer!")
                     errno = MeasureErrorNo.WRONG_ANSWER
     except TVMError as exc:

diff --git a/topi/python/topi/arm_cpu/conv2d.py b/topi/python/topi/arm_cpu/conv2d.py
@@ -211,6 +211,7 @@ def _decl_spatial_pack(cfg, data, kernel, strides, padding, dilation, layout, ou
     output = tvm.compute(oshape, lambda n, co, h, w:
                          conv[n][co//VC][h//VH][w//VW][h%VH][w%VW][co%VC],
                          name='output_unpack', tag='spatial_conv2d_output')
+
     cfg.add_flop(2 * np.prod(oshape) * CI * KH * KW)
     return output