mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8db9dfa2d7
See #127836 for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127846 Approved by: https://github.com/ezyang ghstack dependencies: #127842, #127843, #127844, #127845
23 lines
611 B
Python
23 lines
611 B
Python
# 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
|