Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fixed the rest of Hexagon MS tests
Browse files Browse the repository at this point in the history
masahi committed Oct 7, 2022
1 parent 668633d commit bf6f2da
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions tests/python/contrib/test_hexagon/test_meta_schedule.py
Original file line number Diff line number Diff line change
@@ -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),
)

0 comments on commit bf6f2da

Please sign in to comment.