def _add_deterministic_patch():
"""
Patch torch.utils.deterministic.fill_uninitialized_memory setter so that
setting it also synchronizes the NPU-side option
_npu_fill_uninitialized_memory via torch_npu._C._npu_set_fill_uninitialized_memory.
"""
import torch_npu
import torch.utils.deterministic as det_mod
_Deterministic = det_mod.__class__
_orig_prop = _Deterministic.fill_uninitialized_memory
_orig_fget = _orig_prop.fget
_orig_fset = _orig_prop.fset
def _new_fset(self, mode):
_orig_fset(self, mode)
torch_npu._C._npu_set_fill_uninitialized_memory(mode)
_Deterministic.fill_uninitialized_memory = property(_orig_fget, _new_fset)