diff --git a/tests/python/contrib/test_hexagon/test_meta_schedule.py b/tests/python/contrib/test_hexagon/test_meta_schedule.py index 8b07122c2a17..ab2f75773db0 100644 --- a/tests/python/contrib/test_hexagon/test_meta_schedule.py +++ b/tests/python/contrib/test_hexagon/test_meta_schedule.py @@ -199,7 +199,7 @@ def schedule_dense_for_tune(sch): sch = ms.tune_tir( mod=workload, - target=target, + target=get_hexagon_target("v68"), config=config, work_dir=work_dir, space=ms.space_generator.ScheduleFn(schedule_dense_for_tune), @@ -408,3 +408,7 @@ def test_conv2d_relay_auto_schedule(hexagon_launcher): out = rt_mod.get_output(0).numpy() print(np.max(np.abs(ref - out)), np.mean(np.abs(ref - out))) + + +if __name__ == "__main__": + tvm.testing.main()