From a6b58efe4540fd3d4bbbbc495f1d1dfaddd873ab Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 10 Jun 2025 13:42:43 +0300 Subject: [PATCH] ROCm 6.4 support with --use-nightly --- CHANGELOG.md | 3 ++- installer.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 7ded29486..1593fbf57 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -12,7 +12,8 @@ - don't override user set gfx version with ROCm - **Torch** - - set default to `torch==2.7.1` + - Set default to `torch==2.7.1` + - Support ROCm 6.4 with `---use-nightly` - **SDNQ Quantization** - Add group size support for convolutional layers diff --git a/installer.py b/installer.py index 5eb1a86f3..cf45e3737 100644 --- a/installer.py +++ b/installer.py @@ -669,7 +669,9 @@ def install_rocm_zluda(): os.environ.setdefault('TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL', '1') if args.use_nightly: - if rocm.version is None or float(rocm.version) >= 6.3: # assume the latest if version check fails + if rocm.version is None or float(rocm.version) >= 6.4: # assume the latest if version check fails + torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4') + elif rocm.version == "6.3": torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.3') else: # oldest rocm version on nightly is 6.2.4 torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4')