Fix lint and correct test config option name

Change-Id: I864475c3d03e9b426ce5ef987989216d57f3e019
apache · May 7, 2024 · efb057b · efb057b
1 parent 122bd65
commit efb057b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/python/codegen/test_target_codegen_aarch64.py b/tests/python/codegen/test_target_codegen_aarch64.py
@@ -702,7 +702,7 @@ def before(a: T.handle):
 
 @pytest.mark.skipif(
     llvm_version_major() < 11,
-    reason="Vscale and get.active.lane.mask are not supported in earlier versions of LLVM"
+    reason="Vscale and get.active.lane.mask are not supported in earlier versions of LLVM",
 )
 def test_predicated_scalable_buffer():
     target = "llvm -mtriple=aarch64-linux-gnu -mattr=+sve"

diff --git a/tests/python/tir-transform/test_tir_transform_vectorize.py b/tests/python/tir-transform/test_tir_transform_vectorize.py
@@ -622,7 +622,7 @@ def expected(a: T.handle, b: T.handle):
             )
 
     before_mod = tvm.IRModule.from_expr(before)
-    with tvm.transform.PassContext(config={"tir.enable_buffer_predication": True}):
+    with tvm.transform.PassContext(config={"tir.enable_buffer_level_predication": True}):
         after = tvm.tir.transform.VectorizeLoop()(before_mod)["main"]
     tvm.ir.assert_structural_equal(after, expected)