| 12345678910111213141516171819202122 |
- # mypy: allow-untyped-defs
- import sys
- import types
- import torch
- class _Deterministic(types.ModuleType):
- @property
- 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()
- @fill_uninitialized_memory.setter
- def fill_uninitialized_memory(self, mode):
- return torch._C._set_deterministic_fill_uninitialized_memory(mode)
- sys.modules[__name__].__class__ = _Deterministic
|
