22 lines
584 B
Python
22 lines
584 B
Python
|
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
|