diff --git a/tests/python/unittest/test_auto_scheduler_layout_rewrite.py b/tests/python/unittest/test_auto_scheduler_layout_rewrite.py index 29308ee80fa0c..8bf78067e6a7b 100644 --- a/tests/python/unittest/test_auto_scheduler_layout_rewrite.py +++ b/tests/python/unittest/test_auto_scheduler_layout_rewrite.py @@ -26,6 +26,8 @@ from tvm import topi from tvm import auto_scheduler, te +import os + from test_auto_scheduler_common import get_tiled_matmul, matmul_auto_scheduler_test @@ -73,6 +75,8 @@ def test_correctness_layout_rewrite_rewrite_for_preTransformed(): measure_callbacks=[auto_scheduler.RecordToFile(log_file)], ) auto_scheduler.auto_schedule(task, search_policy, tuning_options) + cmd = "cat %s" % (log_file) + os.system(cmd) inp, _ = auto_scheduler.load_best(log_file, task.workload_key, target) s, bufs = dag.apply_steps_from_state( inp.state, layout_rewrite=auto_scheduler.compute_dag.ComputeDAG.RewriteForPreTransformed @@ -149,6 +153,8 @@ def test_correctness_layout_rewrite_insert_transform_stage(): measure_callbacks=[auto_scheduler.RecordToFile(log_file)], ) auto_scheduler.auto_schedule(task, search_policy, tuning_options) + cmd = "cat %s" % (log_file) + os.system(cmd) inp, _ = auto_scheduler.load_best(log_file, task.workload_key, target) s, bufs = dag.apply_steps_from_state( inp.state, layout_rewrite=auto_scheduler.compute_dag.ComputeDAG.InsertTransformStage