diff --git a/python/tvm/runtime/disco/session.py b/python/tvm/runtime/disco/session.py index 38c4f2a2354c..89ef549df3ee 100644 --- a/python/tvm/runtime/disco/session.py +++ b/python/tvm/runtime/disco/session.py @@ -92,7 +92,7 @@ class DModule(DRef): def __init__(self, dref: DRef, session: "Session") -> None: self.handle = dref.handle - del dref.handle + dref.handle = None self.session = session def __getitem__(self, name: str) -> DPackedFunc: