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()