Skip to content

Commit bba5e32

Browse files
authored
Added env variable to ensure proper behavior in PyTorch with ROCm 7.x (#65)
1 parent b33b7d3 commit bba5e32

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

hpc_launcher/systems/lc/el_capitan_family.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,8 @@ def environment_variables(self) -> list[tuple[str, str]]:
195195
# Ensure that PyTorch respects channel's last for MIOpen (Audited on 1/13/2026)
196196
env_list.append(("PYTORCH_MIOPEN_SUGGEST_NHWC", "1"))
197197
env_list.append(("PYTORCH_MIOPEN_SUGGEST_NHWC_BATCHNORM", "1"))
198+
# Ensure that MIOpen uses Stream-k for PyTorch backwards operations in 7.x (Audited on 3/17/2026)
199+
env_list.append(("TENSILE_SOLUTION_SELECTION_METHOD", "2"))
198200

199201
for i in self._aux_env_list:
200202
env_list.append(i)

0 commit comments

Comments
 (0)