tlc-pack / tvm-tensorir

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[BUG][MetaSchedule] Meta Schedule Layout Rewrite Network Test Error

zxybazh opened this issue · comments

Got the following test error message when running test_meta_schedule_layout_rewrite_network.py.

conv2d NHWC layout is not optimized for x86 with autotvm.
Exception in thread Thread-1:
Traceback (most recent call last):
  File "/home/zxybazh/anaconda3/lib/python3.8/threading.py", line 932, in _bootstrap_inner
    self.run()
  File "/home/zxybazh/anaconda3/lib/python3.8/threading.py", line 870, in run
    self._target(*self._args, **self._kwargs)
  File "/home/zxybazh/tvm-tensorir/python/tvm/meta_schedule/relay_integration.py", line 75, in call_all_topi_funcs
    from tvm.relay.backend import graph_runtime_codegen
ImportError: cannot import name 'graph_runtime_codegen' from 'tvm.relay.backend' (/home/zxybazh/tvm-tensorir/python/tvm/relay/backend/__init__.py)
[16:20:53] /home/zxybazh/tvm-tensorir/src/meta_schedule/database.cc:157: Loading tuning records from: conv2d.json
[16:20:53] /home/zxybazh/tvm-tensorir/src/meta_schedule/database.cc:183: Nothing is loaded. File does not exist or cannot be opened
Traceback (most recent call last):
  File "test_meta_schedule_layout_rewrite_network.py", line 253, in <module>
    test_conv2d()
  File "test_meta_schedule_layout_rewrite_network.py", line 231, in test_conv2d
    tune_and_check("conv2d.json", mod, data, weight)
  File "test_meta_schedule_layout_rewrite_network.py", line 178, in tune_and_check
    lib = relay.build_module.build(mod, TARGET, params={"weight": weight}, tune_result={})
TypeError: build() got an unexpected keyword argument 'tune_result'

Won't fix