[Draft] Update

apache · Nov 4, 2020 · 56b6e39 · 56b6e39
1 parent c4c95c4
commit 56b6e39
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/unittest/test_auto_scheduler_layout_rewrite.py b/tests/python/unittest/test_auto_scheduler_layout_rewrite.py
@@ -169,7 +169,9 @@ def test_correctness_layout_rewrite_insert_transform_stage():
         np_args = [np.random.randn(*topi.get_const_tuple(x.shape)).astype(x.dtype) for x in bufs]
 
         func = tvm.build(s, bufs, target=target)
+        print(func.get_source())
         func_ref = tvm.build(s_ref, bufs_ref, target=target)
+        print(func_ref.get_source())
 
         ctx = tvm.context(str(target))
         ctx_ref = tvm.cpu()
@@ -197,4 +199,4 @@ def ffff():
     # test_apply_steps_with_layout_rewrite()
     test_correctness_layout_rewrite_rewrite_for_preTransformed()
     test_correctness_layout_rewrite_insert_transform_stage()
-    ffff()
+    # ffff()