pytorch

1
# mypy: allow-untyped-defs
2
import sys
3
import types
4

5
import torch
6

7

8
class _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

22
sys.modules[__name__].__class__ = _Deterministic
23