You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
584 B
22 lines
584 B
5 months ago
|
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
|