Skip to content

Commit

Permalink
[PYTHON] Enable proper error message in python package (#7521)
Browse files Browse the repository at this point in the history
  • Loading branch information
tqchen authored Feb 25, 2021
1 parent b77c019 commit 772fa6b
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion python/tvm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,11 @@
from .contrib import rocm as _rocm, nvcc as _nvcc, sdaccel as _sdaccel


# NOTE: This file should be python2 compatible so we can
# raise proper error message when user run the package using
# an older version of the python


def _should_print_backtrace():
in_pytest = "PYTEST_CURRENT_TEST" in os.environ
tvm_backtrace = os.environ.get("TVM_BACKTRACE", "0")
Expand All @@ -76,7 +81,7 @@ def _should_print_backtrace():
tvm_backtrace = bool(int(tvm_backtrace))
except ValueError:
raise ValueError(
f"invalid value for TVM_BACKTRACE `{tvm_backtrace}`, please set to 0 or 1."
"invalid value for TVM_BACKTRACE {}, please set to 0 or 1.".format(tvm_backtrace)
)

return in_pytest or tvm_backtrace
Expand Down

0 comments on commit 772fa6b

Please sign in to comment.