Adding all project files
This commit is contained in:
parent
6c9e127bdc
commit
cd4316ad0f
42289 changed files with 8009643 additions and 0 deletions
22
venv/Lib/site-packages/torch/utils/deterministic.py
Normal file
22
venv/Lib/site-packages/torch/utils/deterministic.py
Normal file
|
@ -0,0 +1,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
|
Loading…
Add table
Add a link
Reference in a new issue