mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 09:38:23 +02:00
[MIOPEN} Set MIOPEN_FIND_MODE 2 & enable GFX1200
This commit is contained in:
@@ -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')
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user