From 074f48af5efed6850707db3a7becdc80c84d1527 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Fri, 7 Oct 2022 12:44:22 -0700 Subject: [PATCH] fixed the rest of Hexagon MS tests --- .../contrib/test_hexagon/test_meta_schedule.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/python/contrib/test_hexagon/test_meta_schedule.py b/tests/python/contrib/test_hexagon/test_meta_schedule.py index 18cc88cfa5b7..6e12f4b205d1 100644 --- a/tests/python/contrib/test_hexagon/test_meta_schedule.py +++ b/tests/python/contrib/test_hexagon/test_meta_schedule.py @@ -205,7 +205,7 @@ def schedule_dense_for_tune(sch): schedule_dense_for_tune, sch_rules=[], postprocs=[], - mutator_probs=[], + mutator_probs={}, ), strategy="replay-trace", builder=get_hexagon_local_builder(), @@ -307,6 +307,7 @@ def test_vrmpy_dense_auto_tensorize(hexagon_launcher): postproc.RewriteTensorize(vectorize_init_loop=True), ] + # Make this to False to compile and run the best tuned schedule if True: with tempfile.TemporaryDirectory() as work_dir: target = get_hexagon_target("v68") @@ -315,10 +316,13 @@ def test_vrmpy_dense_auto_tensorize(hexagon_launcher): target=target, max_trials_global=8, num_trials_per_iter=8, - max_trials_per_task=8, work_dir=work_dir, - sch_rules=lambda: sch_rules, - postprocs=lambda: postprocs, + space=ms.space_generator.PostOrderApply( + f_block_filter=None, + sch_rules=sch_rules, + postprocs=postprocs, + mutator_probs={}, + ), builder=get_hexagon_local_builder(), runner=get_hexagon_rpc_runner(hexagon_launcher, number=10), )