File size: 611 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# 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
|