update unittests

apache · Feb 4, 2023 · d6816c4 · d6816c4
1 parent 9c673ba
commit d6816c4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/src/tir/schedule/error.cc b/src/tir/schedule/error.cc
@@ -31,6 +31,7 @@ String ScheduleError::RenderReport(const String& primitive) const {
   int n_locs = locs.size();
   std::string msg = DetailRenderTemplate();
   PrinterConfig cfg;
+  cfg->syntax_sugar = false;
   if (n_locs > 0) {
     for (int i = 0; i < n_locs; ++i) {
       std::string name = locs[i]->GetTypeKey() + '#' + std::to_string(i);

diff --git a/tests/python/unittest/test_tvmscript_error_report.py b/tests/python/unittest/test_tvmscript_error_report.py
@@ -459,9 +459,10 @@ def test_reorder_fail_block():
     with pytest.raises(tvm.tir.ScheduleError) as execinfo:
         sch.reorder(l, i)
     expected_sub_error_message = (
-        "            # tir.Block#0\n"
-        '            with T.block("B"):\n'
-        "            ^^^^^^^^^^^^^^^^^^\n"
+        "                            # tir.Block#0\n"
+        "                            ^^^^^^^^^^^^^\n"
+        '                            with T.block("B"):\n'
+        "^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n"
     )
     assert expected_sub_error_message in str(execinfo.value)
 
@@ -473,10 +474,11 @@ def test_reorder_fail_nested_loop_inner():
     with pytest.raises(tvm.tir.ScheduleError) as execinfo:
         sch.reorder(k, i)
     expected_sub_error_message = (
-        "        for i in T.serial(128):\n"
-        "            # tir.For#0\n"
-        "            for j in T.serial(128):\n"
-        "            ^^^^^^^^^^^^^^^^^^^^^^^\n"
+        "            for i in range(128):\n"
+        "                # tir.For#0\n"
+        "                ^^^^^^^^^^^\n"
+        "                for j in range(128):\n"
+        "^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n"
     )
     assert expected_sub_error_message in str(execinfo.value)
 
@@ -488,10 +490,11 @@ def test_fuse_fail_nested_loop_outer():
     with pytest.raises(tvm.tir.ScheduleError) as execinfo:
         sch.fuse(k, i)
     expected_sub_error_message = (
-        "        # tir.For#1\n"
-        "        for i in T.serial(128):\n"
-        "        ^^^^^^^^^^^^^^^^^^^^^^^\n"
-        "            for j in T.serial(128):\n"
+        "            # tir.For#1\n"
+        "            ^^^^^^^^^^^\n"
+        "            for i in range(128):\n"
+        "^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n"
+        "                for j in range(128):\n"
     )
     assert expected_sub_error_message in str(execinfo.value)
 
@@ -503,8 +506,9 @@ def test_report_error_root_block():
         sch.compute_inline(root)
     expected_sub_error_message = (
         "        # tir.Block#0\n"
+        "        ^^^^^^^^^^^^^\n"
         '        with T.block("root"):\n'
-        "        ^^^^^^^^^^^^^^^^^^^^^\n"
+        "^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n"
     )
     assert expected_sub_error_message in str(execinfo.value)