From 035d0053e53bf5c52ba62a2936b003318656ac87 Mon Sep 17 00:00:00 2001 From: Tristan Konolige Date: Wed, 13 Jan 2021 21:02:35 -0800 Subject: [PATCH] [FIX] Remove leftovers from check_correctness (#7272) * [FIX] Remove leftovers from check_correctness * remove unused numpy import --- python/tvm/autotvm/measure/measure_methods.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/python/tvm/autotvm/measure/measure_methods.py b/python/tvm/autotvm/measure/measure_methods.py index f6f0e429a22ab..cb801ba728721 100644 --- a/python/tvm/autotvm/measure/measure_methods.py +++ b/python/tvm/autotvm/measure/measure_methods.py @@ -31,8 +31,6 @@ from collections import namedtuple import tempfile -import numpy as np - import tvm._ffi import tvm.ir.transform from tvm import nd, rpc as _rpc @@ -578,13 +576,6 @@ def run_through_rpc( costs = list(costs) costs.sort() costs = tuple(costs[1:-1]) - - # 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): - logger.warning("Wrong Answer!") - errno = MeasureErrorNo.WRONG_ANSWER except TVMError as exc: msg = str(exc) if "Stack trace returned" in msg: