Skip to content

Commit 56005d2

Browse files
authored
HPU accelerator memory mapping is broken because of torch fill uninit memory (#7209)
1 parent 29fa95a commit 56005d2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

accelerator/hpu_accelerator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@ def __init__(self):
2323
import habana_frameworks.torch.hpu as hpu
2424
self.hpu = hpu
2525
torch.use_deterministic_algorithms(True)
26+
# TODO: remove this WA when memory mapping break is resolved.
27+
torch.utils.deterministic.fill_uninitialized_memory = False
2628
except ImportError as e:
2729
raise ValueError(
2830
f"HPU_Accelerator requires habana_frameworks.torch.hpu, which is not installed on this system.")

0 commit comments

Comments
 (0)