From 4c435fddb5f0971a7d6bc492189128d4612729f4 Mon Sep 17 00:00:00 2001
From: Bratzmeister <bratzmeister@proton.me>
Date: Mon, 11 Nov 2024 20:01:43 +0100
Subject: [PATCH] fix --cuda-device arg for AMD/HIP devices

CUDA_VISIBLE_DEVICES is ignored for HIP devices/backend. Instead it uses HIP_VISIBLE_DEVICES. Setting this environment variable has no side effect for CUDA/NVIDIA so it can safely be set in any case and vice versa.
---
 main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/main.py b/main.py
index c23210861..392242902 100644
--- a/main.py
+++ b/main.py
@@ -70,7 +70,9 @@ if os.name == "nt":
 
 if __name__ == "__main__":
     if args.cuda_device is not None:
+        if 
         os.environ['CUDA_VISIBLE_DEVICES'] = str(args.cuda_device)
+        os.environ['HIP_VISIBLE_DEVICES'] = str(args.cuda_device)
         logging.info("Set cuda device to: {}".format(args.cuda_device))
 
     if args.deterministic: