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