Revert evil work around.

junrushao · Jan 13, 2022 · 214db44 · 214db44
1 parent 1ee1c68
commit 214db44
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/meta_schedule/space_generator/post_order_apply.cc b/src/meta_schedule/space_generator/post_order_apply.cc
@@ -98,7 +98,7 @@ class PostOrderApplyNode : public SpaceGeneratorNode {
     tir::Schedule sch = tir::Schedule::Traced(        //
         /*mod=*/mod_,                                 //
         /*rand_state=*/ForkSeed(&this->rand_state_),  //
-        /*debug_mode=*/0,  // tir::kVerifySRefTree | tir::kVerifyCachedFlags
+        /*debug_mode=*/tir::kVerifySRefTree | tir::kVerifyCachedFlags,
         /*error_render_level=*/tir::ScheduleErrorRenderLevel::kDetail);
 
     std::vector<ScheduleAndUnvisitedBlocks> stack;

diff --git a/tests/python/unittest/test_meta_schedule_tune_relay.py b/tests/python/unittest/test_meta_schedule_tune_relay.py
@@ -30,8 +30,8 @@
 logging.getLogger("tvm.meta_schedule").setLevel(logging.DEBUG)
 
 
-@pytest.mark.skip("Integration test")
-@pytest.mark.parametrize("model_name", ["bert_base"])
+# @pytest.mark.skip("Integration test")
+@pytest.mark.parametrize("model_name", ["resnet18", "mobilenet_v2", "bert_base"])
 @pytest.mark.parametrize("batch_size", [1])
 @pytest.mark.parametrize("target", ["llvm --num-cores=16", "nvidia/geforce-rtx-3070"])
 def test_meta_schedule_tune_relay(model_name: str, batch_size: int, target: str):