upgrade

Author: lshqqytiger

modules/devices.py

@@ -3,7 +3,7 @@
 from functools import lru_cache
 
 import torch
-from modules import errors, shared, npu_specific
+from modules import errors, shared, npu_specific, rocm
 
 if sys.platform == "darwin":
     from modules import mac_specific
@@ -34,6 +34,10 @@ def has_zluda() -> bool:
         return False
 
 
+def get_hip_agent() -> rocm.Agent:
+    return rocm.Agent(device)
+
+
 def cuda_no_autocast(device_id=None) -> bool:
     if device_id is None:
         device_id = get_cuda_device_id()

modules/launch_utils.py

@@ -38,9 +38,9 @@ def check_python_version():
     micro = sys.version_info.micro
 
     if is_windows:
-        supported_minors = [10, 11]
+        supported_minors = [11, 12]
     else:
-        supported_minors = [7, 8, 9, 10, 11]
+        supported_minors = [10, 11, 12]
 
     if not (major == 3 and minor in supported_minors):
         import modules.errors
@@ -564,7 +564,7 @@ def prepare_environment():
             if device.therock is None:
                 backend = "zluda"
             else:
-                run_pip(f"install rocm rocm-sdk-core --index-url https://rocm.nightlies.amd.com/v2-staging/{device.therock}", "rocm")
+                run_pip(f"install rocm rocm-sdk-core --index-url https://rocm.nightlies.amd.com/{device.therock}", "rocm")
                 rocm.refresh()
 
         msg = f'ROCm: version={rocm.version}'
@@ -575,7 +575,7 @@ def prepare_environment():
         if backend == "rocm":
             if sys.platform == "win32":
                 if isinstance(rocm.environment, rocm.PythonPackageEnvironment):
-                    torch_command = os.environ.get('TORCH_COMMAND', f'pip install torch torchvision --index-url https://rocm.nightlies.amd.com/v2-staging/{device.therock}')
+                    torch_command = os.environ.get('TORCH_COMMAND', f'pip install torch torchvision --index-url https://rocm.nightlies.amd.com/{device.therock}')
                 else:
                     torch_command = os.environ.get('TORCH_COMMAND', 'pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-6.4.4/torch-2.8.0a0%2Bgitfc14c65-cp312-cp312-win_amd64.whl https://repo.radeon.com/rocm/windows/rocm-rel-6.4.4/torchvision-0.24.0a0%2Bc85f008-cp312-cp312-win_amd64.whl')
             else: