Revert "[Unity] Implement relax.transform.KillAfterLastUse (#15810)"

python/tvm/relax/transform/transform.py

@@ -388,16 +388,6 @@ def LowerAllocTensor() -> tvm.ir.transform.Pass:
     return _ffi_api.LowerAllocTensor()  # type: ignore
 
 
-def KillAfterLastUse() -> tvm.ir.transform.Pass:
-    """Drop all tensor/storage objects after last use
-
-    Returns
-    -------
-    ret : tvm.ir.transform.Pass
-    """
-    return _ffi_api.KillAfterLastUse()  # type: ignore
-
-
 def VMBuiltinLower() -> tvm.ir.transform.Pass:
     """Lowering generic intrinsic to VM intrinsics.
 

python/tvm/relax/vm_build.py

@@ -315,7 +315,6 @@ def foo(x: Tensor((3, 4), "float32"), y: Tensor((3, 4), "float32")):
         passes.append(relax.transform.RewriteCUDAGraph())
 
     passes.append(relax.transform.LowerAllocTensor())
-    passes.append(relax.transform.KillAfterLastUse())
 
     passes.append(relax.transform.VMBuiltinLower())
     passes.append(relax.transform.VMShapeLower())

src/relax/transform/utils.h

@@ -388,19 +388,6 @@ inline String GetCodegenName(const std::string& composite_name) {
  */
 Expr EliminateCommonSubexpr(const Expr& expr, bool call_only = false);
 
-/* \brief Remove use of trivial bindings
- *
- * Utility for simplifying relax expressions by folding var bindings
- * and match shape nodes.  May include other forms of simplification
- * in the future.  Ideally should be used before constant folding and
- * eliminating unused bindings.
- *
- * \param expr The expression to be canonicalized
- *
- * \ret The canonicalized expression
- */
-Expr CanonicalizeBindings(const Expr& expr);
-
 }  // namespace relax
 }  // namespace tvm