ROCm set MIOPEN_FIND_ENFORCE to SEARCH

This commit is contained in:
Disty0
2025-05-18 16:36:27 +03:00
parent 3d8390de9b
commit 7f2d77e956
+2
View File
@@ -667,6 +667,8 @@ def install_rocm_zluda():
if os.environ.get("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", None) is None:
os.environ.setdefault('TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL', '1')
if os.environ.get("MIOPEN_FIND_ENFORCE", None) is None:
os.environ.setdefault('MIOPEN_FIND_ENFORCE', 'SEARCH')
if args.use_nightly:
if rocm.version is None or float(rocm.version) >= 6.3: # assume the latest if version check fails