# mypy: allow-untyped-defs | |
import sys | |
import types | |
import torch | |
class _Deterministic(types.ModuleType): | |
def fill_uninitialized_memory(self): | |
""" | |
Whether to fill uninitialized memory with a known value when | |
:meth:`torch.use_deterministic_algorithms()` is set to ``True``. | |
""" | |
return torch._C._get_deterministic_fill_uninitialized_memory() | |
def fill_uninitialized_memory(self, mode): | |
return torch._C._set_deterministic_fill_uninitialized_memory(mode) | |
sys.modules[__name__].__class__ = _Deterministic | |