def __enter__(self):
if hasattr(self, "inner"):
raise RuntimeError(
f"{self} has already been used as a mode, please create and use a fresh version"
)
old = _get_torch_dispatch_mode()
self.inner = old
_set_torch_dispatch_mode(self)
def __enter__(self):
old = _get_torch_dispatch_mode()
if hasattr(self, "inner"):
raise RuntimeError(f"{self} has already been used as a mode. Please use a fresh version or use restore")
else:
self.inner = old
if old is None:
self.ancestors = set()
else:
self.ancestors = self.inner.ancestors.union({self.inner})
_set_torch_dispatch_mode(self)
return self
def get_mode(self):
return _get_torch_dispatch_mode()
