1# mypy: allow-untyped-defs 2import sys 3import types 4 5import torch 6 7 8class _Deterministic(types.ModuleType): 9 @property 10 def fill_uninitialized_memory(self): 11 """ 12 Whether to fill uninitialized memory with a known value when 13 :meth:`torch.use_deterministic_algorithms()` is set to ``True``. 14 """ 15 return torch._C._get_deterministic_fill_uninitialized_memory() 16 17 @fill_uninitialized_memory.setter 18 def fill_uninitialized_memory(self, mode): 19 return torch._C._set_deterministic_fill_uninitialized_memory(mode) 20 21 22sys.modules[__name__].__class__ = _Deterministic 23