From 9647a607c11822d224e037378f140116eb5701fb Mon Sep 17 00:00:00 2001
From: Hameer Abbasi <hameerabbasi@yahoo.com>
Date: Sun, 23 Jul 2023 10:24:41 +0200
Subject: [PATCH] ROCm-related fixes

1. Don't set `HSA_OVERRIDE_GFX_VERSION=10.3.0` by default.
   a. A badly-detected default value usually means a broken ROCm install.
   b. Causes issues with older GPUs.
2. Set `TENSORFLOW_PACKAGE=tensorflow-rocm`, as the default version doesn't contain GPU binaries for ROCm.
---
 installer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/installer.py b/installer.py
index ea8914634..1e7bda491 100644
--- a/installer.py
+++ b/installer.py
@@ -311,8 +311,8 @@ def check_torch():
         xformers_package = os.environ.get('XFORMERS_PACKAGE', 'xformers==0.0.20' if opts.get('cross_attention_optimization', '') == 'xFormers' else 'none')
     elif allow_rocm and (shutil.which('rocminfo') is not None or os.path.exists('/opt/rocm/bin/rocminfo') or os.path.exists('/dev/kfd')):
         log.info('AMD ROCm toolkit detected')
-        os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', '10.3.0')
         os.environ.setdefault('PYTORCH_HIP_ALLOC_CONF', 'garbage_collection_threshold:0.8,max_split_size_mb:512')
+        os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow-rocm')
         torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/rocm5.4.2')
         xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
     elif allow_ipex and (args.use_ipex or shutil.which('sycl-ls') is not None or os.environ.get('ONEAPI_ROOT') is not None or os.path.exists('/opt/intel/oneapi')):