Fix backend.

apache · Jun 23, 2022 · b38bf33 · b38bf33
1 parent 0f273b9
commit b38bf33
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/python/tvm/auto_scheduler/testing/tune_onnx.py b/python/tvm/auto_scheduler/testing/tune_onnx.py
@@ -227,6 +227,7 @@ def main():
         dev_type=ARGS.target.kind.name,
         args=input_data,
         continuation=f_timer(ARGS.backend),
+        backend=ARGS.backend,
     )
 
 

diff --git a/python/tvm/auto_scheduler/testing/tune_relay.py b/python/tvm/auto_scheduler/testing/tune_relay.py
@@ -228,6 +228,7 @@ def main():
         dev_type=ARGS.target.kind.name,
         args=input_data,
         continuation=f_timer(ARGS.backend),
+        backend=ARGS.backend,
     )
 
 

diff --git a/python/tvm/meta_schedule/testing/tune_onnx.py b/python/tvm/meta_schedule/testing/tune_onnx.py
@@ -182,6 +182,7 @@ def main():
         dev_type=ARGS.target.kind.name,
         args=input_data,
         continuation=f_timer(ARGS.backend),
+        backend=ARGS.backend,
     )
 
 

diff --git a/python/tvm/meta_schedule/testing/tune_relay.py b/python/tvm/meta_schedule/testing/tune_relay.py
@@ -184,6 +184,7 @@ def main():
         dev_type=ARGS.target.kind.name,
         args=input_data,
         continuation=f_timer(ARGS.backend),
+        backend=ARGS.backend,
     )