From b38bf338904bde9b198e6a92c9962b839dc1375e Mon Sep 17 00:00:00 2001 From: Xiyou Zhou Date: Wed, 22 Jun 2022 22:41:02 -0700 Subject: [PATCH] Fix backend. --- python/tvm/auto_scheduler/testing/tune_onnx.py | 1 + python/tvm/auto_scheduler/testing/tune_relay.py | 1 + python/tvm/meta_schedule/testing/tune_onnx.py | 1 + python/tvm/meta_schedule/testing/tune_relay.py | 1 + 4 files changed, 4 insertions(+) diff --git a/python/tvm/auto_scheduler/testing/tune_onnx.py b/python/tvm/auto_scheduler/testing/tune_onnx.py index 260d5814cd6aa..6608a6993341e 100644 --- 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 index 4c9eaa985c4ae..0d9912e004dc5 100644 --- 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 index b3737b6f99be9..463f55520ec8d 100644 --- 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 index 3afeb80f71b06..ee168d3bb380c 100644 --- 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, )