Skip to content

Commit

Permalink
[FIX] Remove leftovers from check_correctness (apache#7272)
Browse files Browse the repository at this point in the history
* [FIX] Remove leftovers from check_correctness

* remove unused numpy import
  • Loading branch information
tkonolige authored and trevor-m committed Jan 21, 2021
1 parent 3de9d48 commit 4d50d59
Showing 1 changed file with 0 additions and 9 deletions.
9 changes: 0 additions & 9 deletions python/tvm/autotvm/measure/measure_methods.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 4d50d59

Please sign in to comment.