[Draft]

apache · Nov 3, 2020 · 1527e4d · 1527e4d
1 parent 3222306
commit 1527e4d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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