MCPcopy
hub / github.com/state-spaces/mamba / use_deterministic_mode

Function use_deterministic_mode

mamba_ssm/utils/determinism.py:21–27  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

19
20
21def use_deterministic_mode():
22 if _deterministic_override is not None:
23 return _deterministic_override
24 env = os.environ.get('MAMBA_DETERMINISTIC')
25 if env:
26 return env[0] == '1'
27 return torch.are_deterministic_algorithms_enabled()
28
29
30def set_deterministic_mode(value):

Callers 8

_chunk_scan_bwd_dCFunction · 0.90
_chunk_scan_bwd_dxFunction · 0.90
_chunk_cumsum_bwdFunction · 0.90
_chunk_state_bwd_dxFunction · 0.90
_chunk_state_bwd_dbFunction · 0.90
autotune_configsFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected