[Fix] Remove redundant "remove_all_unused" in IPC memory lowering

This commit removes the redundant invocation of `remove_all_unused`
function in the GPU IPC memory allocation lowering pass. This is
because the pass only mutates one call at a time, and thus will
not introduce new unused bindings.

python/tvm/relax/transform/ipc_allreduce_rewrite.py

@@ -138,7 +138,6 @@ def transform(self) -> IRModule:
         for g_var, func in self.mod.functions_items():
             if isinstance(func, relax.Function):
                 updated_func = self.visit_expr(func)
-                updated_func = remove_all_unused(updated_func)
                 self.builder_.update_func(g_var, updated_func)
         return self.builder_.get()
 

python/tvm/relax/transform/lower_gpu_ipc_alloc_storage.py

@@ -49,7 +49,6 @@ def transform(self) -> IRModule:
         for g_var, func in self.mod.functions_items():
             if isinstance(func, relax.Function):
                 updated_func = self.visit_expr(func)
-                updated_func = remove_all_unused(updated_func)
                 self.builder_.update_func(g_var, updated_func)
         return self.builder_.get()