Skip to content

Commit

Permalink
change import so must use tvm.driver.tvmc instead of tvm.tvmc
Browse files Browse the repository at this point in the history
  • Loading branch information
Jocelyn committed Mar 23, 2021
1 parent eba7d63 commit 250efd1
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 10 deletions.
1 change: 0 additions & 1 deletion python/tvm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,6 @@

# tvm.driver
from .driver import build, lower
from .driver import tvmc

# tvm.parser
from . import parser
Expand Down
10 changes: 1 addition & 9 deletions tests/python/driver/tvmc/test_frontends.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@

from tvm.ir.module import IRModule

from tvm import tvmc
from tvm.driver import tvmc
from tvm.driver.tvmc.common import TVMCException


Expand Down Expand Up @@ -137,16 +137,8 @@ def verify_load_model__onnx(model, **kwargs):
def test_load_model__onnx(onnx_resnet50):
# some CI environments wont offer onnx, so skip in case it is not present
pytest.importorskip("onnx")
<<<<<<< HEAD
mod, params = verify_load_model__onnx(onnx_resnet50)
# check whether one known value is part of the params dict
=======

mod, params = tvmc.load(onnx_resnet50)
assert type(mod) is IRModule
assert type(params) is dict
## check whether one known value is part of the params dict
>>>>>>> adjust tests to new imports
assert "resnetv24_batchnorm0_gamma" in params.keys()
mod, params = verify_load_model__onnx(onnx_resnet50, freeze_params=True)
# check that the parameter dict is empty, implying that they have been folded into constants
Expand Down

0 comments on commit 250efd1

Please sign in to comment.