fix bugs

Author: tqchen

python/tvm/tir/block_dependence_info.py

@@ -51,7 +51,7 @@ def __init__(self, mod: Union[IRModule, PrimFunc]):
         if not isinstance(mod, IRModule):
             raise TypeError(f"Expected `mod` to be PrimFunc or IRModule, but gets: {mod}")
         self.__init_handle_by_constructor__(
-            _ffi_api.BlockDependenceInfo,  # type: ignore # pylint: disable=no-member
+            _ffi_api.SBlockDependenceInfo,  # type: ignore # pylint: disable=no-member
             mod,
         )
 
@@ -68,7 +68,7 @@ def get_sref(self, block: SBlock) -> Optional[StmtSRef]:
         sref : StmtSRef
             The corresponding sref
         """
-        return _ffi_api.BlockDependenceInfoGetSRef(self, block)  # type: ignore # pylint: disable=no-member
+        return _ffi_api.SBlockDependenceInfoGetSRef(self, block)  # type: ignore # pylint: disable=no-member
 
     def get_sblock_scope(self, block_sref: StmtSRef) -> SBlockScope:
         """Get the SBlockScope correpsonding to the block sref
@@ -83,6 +83,6 @@ def get_sblock_scope(self, block_sref: StmtSRef) -> SBlockScope:
         scope : StmtSRef
             The corresponding SBlockScope
         """
-        return _ffi_api.BlockDependenceInfoGetSBlockScope(  # type: ignore # pylint: disable=no-member
+        return _ffi_api.SBlockDependenceInfoGetSBlockScope(  # type: ignore # pylint: disable=no-member
             self, block_sref
         )

tests/python/tir-base/test_tir_block_dependence_info.py

@@ -24,7 +24,7 @@
 from tvm import tir
 from tvm.ir import IRModule
 from tvm.script import tir as T
-from tvm.tir import PrimFunc, BlockDependenceInfo
+from tvm.tir import PrimFunc, SBlockDependenceInfo
 from tvm.tir.stmt_functor import post_order_visit
 from tvm.tir.block_scope import DepKind
 
@@ -128,21 +128,21 @@ def _get_dependency_kind_name(dep_kind):
 
 def test_RAW_dependences():
     func = elementwise
-    dependence_info = BlockDependenceInfo(func)
+    dependence_info = SBlockDependenceInfo(func)
     blocks = get_sblocks(func)
     _verify_dependence(dependence_info, blocks["B"], blocks["C"], DepKind.RAW)
 
 
 def test_WAR_dependences():
     func = war_dependency
-    dependence_info = BlockDependenceInfo(func)
+    dependence_info = SBlockDependenceInfo(func)
     blocks = get_sblocks(func)
     _verify_dependence(dependence_info, blocks["C"], blocks["B"], DepKind.WAR)
 
 
 def test_RAW_and_WAW_dependences():
     func = matmul
-    dependence_info = BlockDependenceInfo(func)
+    dependence_info = SBlockDependenceInfo(func)
     blocks = get_sblocks(func)
     _verify_dependence(dependence_info, blocks["init"], blocks["update"], DepKind.RAW)
     _verify_dependence(dependence_info, blocks["init"], blocks["update"], DepKind.WAW)

tests/python/tir-base/test_tir_renew_defs.py

@@ -20,7 +20,7 @@
 from tvm.script import tir as T
 from tvm.tir.buffer import Buffer
 from tvm.tir.function import PrimFunc
-from tvm.tir.stmt import Block
+from tvm.tir.stmt import SBlock
 
 
 def _check_func_signature_remap(lhs: PrimFunc, rhs: PrimFunc):