[MIOPEN} Set MIOPEN_FIND_MODE 2 & enable GFX1200

This commit is contained in:
Crashingalexsan
2026-01-27 16:40:27 -06:00
parent df3ff41ebb
commit eeb176c0d0
2 changed files with 1 additions and 4 deletions
+1
View File
@@ -1447,6 +1447,7 @@ def set_environment():
os.environ.setdefault('TORCH_CUDNN_V8_API_ENABLED', '1')
os.environ.setdefault('TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD', '1')
os.environ.setdefault('TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL', '1')
os.environ.setdefault('MIOPEN_FIND_MODE', '2')
os.environ.setdefault('UR_L0_ENABLE_RELAXED_ALLOCATION_LIMITS', '1')
os.environ.setdefault('USE_TORCH', '1')
os.environ.setdefault('UV_INDEX_STRATEGY', 'unsafe-any-match')
-4
View File
@@ -305,10 +305,6 @@ if sys.platform == "win32":
log.debug(f'ROCm: selected={agents}')
if not agent.blaslt_supported:
log.warning(f'ROCm: hipBLASLt unavailable agent={agent}')
if (agent.gfx_version & 0xFFF0) == 0x1200:
# disable MIOpen for gfx120x
torch.backends.cudnn.enabled = False
log.debug('ROCm: disabled MIOpen')
if sys.platform == "win32":
apply_triton_patches()