Refine preview and EOL gating

Author: Godnight1006

API.md

@@ -15,7 +15,7 @@ torchruntime.install(["torch", "torchvision<0.20"])
 
 Optional flags:
 - `preview=True` to allow preview builds (e.g. ROCm 6.4, nightly builds, XPU test index)
-- `unsupported=False` to forbid EOL/unsupported builds (errors instead of installing older builds)
+- `unsupported=False` to forbid EOL/unsupported builds (e.g. Torch-DirectML, IPEX, Torch 1.x)
 
 On Windows CUDA, Linux ROCm (6.x+), and Linux XPU, this also installs the appropriate Triton package to enable `torch.compile` (`triton-windows`, `pytorch-triton-rocm`, or `pytorch-triton-xpu`).
 

tests/test_platform_detection.py

@@ -36,7 +36,8 @@ def test_amd_gpu_navi4_linux(monkeypatch):
         with pytest.raises(NotImplementedError):
             get_torch_platform(gpu_infos)
     else:
-        assert get_torch_platform(gpu_infos) == "rocm6.2"
+        assert get_torch_platform(gpu_infos) == "cpu"
+        assert get_torch_platform(gpu_infos, preview=True) == "rocm6.4"
 
 
 def test_amd_gpu_navi3_linux(monkeypatch, capsys):

tests/test_segmentation.py

@@ -1,6 +1,6 @@
 import pytest
 from torchruntime.device_db import GPU
-from torchruntime.platform_detection import AMD, INTEL, NVIDIA, get_torch_platform, py_version
+from torchruntime.platform_detection import AMD, INTEL, get_torch_platform, py_version
 
 
 def test_preview_rocm_6_4_selection(monkeypatch):
@@ -11,25 +11,21 @@ def test_preview_rocm_6_4_selection(monkeypatch):
     if py_version < (3, 9):
         pytest.skip("Navi 4 requires Python 3.9+")
 
-    # Default: preview=False -> rocm6.2
-    assert get_torch_platform(gpu_infos) == "rocm6.2"
-    assert get_torch_platform(gpu_infos, preview=False) == "rocm6.2"
+    # Default: preview=False -> cpu
+    assert get_torch_platform(gpu_infos) == "cpu"
+    assert get_torch_platform(gpu_infos, preview=False) == "cpu"
 
     # preview=True -> rocm6.4
     assert get_torch_platform(gpu_infos, preview=True) == "rocm6.4"
 
 
-def test_eol_cu118_selection(monkeypatch):
-    monkeypatch.setattr("torchruntime.platform_detection.os_name", "Windows")
-    monkeypatch.setattr("torchruntime.platform_detection.arch", "amd64")
-    # Kepler architecture (e.g. GTX 780)
-    gpu_infos = [GPU(NVIDIA, "NVIDIA", "1004", "GK110 [GeForce GTX 780]", True)]
+def test_eol_rocm_5_2_selection(monkeypatch):
+    monkeypatch.setattr("torchruntime.platform_detection.os_name", "Linux")
+    monkeypatch.setattr("torchruntime.platform_detection.arch", "x86_64")
+    gpu_infos = [GPU(AMD, "AMD", 0x1234, "Navi 10", True)]
 
-    # Default: unsupported=True -> cu118
-    assert get_torch_platform(gpu_infos) == "cu118"
-    assert get_torch_platform(gpu_infos, unsupported=True) == "cu118"
+    assert get_torch_platform(gpu_infos) == "rocm5.2"
 
-    # unsupported=False -> raises ValueError
     with pytest.raises(ValueError, match="considered End-of-Life"):
         get_torch_platform(gpu_infos, unsupported=False)
 

torchruntime/__main__.py

@@ -38,7 +38,7 @@ def print_usage(entry_command: str):
 Options:
     --uv               Use uv instead of pip for installation
     --preview          Allow preview builds (e.g. ROCm 6.4)
-    --no-unsupported   Forbid EOL/unsupported builds (e.g. CUDA 11.8)
+    --no-unsupported   Forbid EOL/unsupported builds (e.g. DirectML / IPEX / Torch 1.x)
 
 Version specification formats (follows pip format):
     package==2.1.0     Exact version

torchruntime/platform_detection.py

@@ -103,7 +103,7 @@ def get_torch_platform(gpu_infos, packages=[], preview=False, unsupported=True):
         platform = _get_platform_for_integrated(integrated_devices, preview=preview)
 
     # Segmentation Logic
-    EOL_PLATFORMS = {"cu118", "directml", "ipex", "rocm5.7", "rocm5.5", "rocm5.2", "rocm4.2"}
+    EOL_PLATFORMS = {"directml", "ipex", "rocm5.2", "rocm4.2"}
 
     if not unsupported and platform in EOL_PLATFORMS:
         raise ValueError(
@@ -138,7 +138,13 @@ def _get_platform_for_discrete(gpu_infos, packages=None, preview=False):
                     raise NotImplementedError(
                         f"Torch does not support Navi 4x series of GPUs on Python 3.8. Please switch to a newer Python version to use the latest version of torch!"
                     )
-                return "rocm6.4" if preview else "rocm6.2"
+                if preview:
+                    return "rocm6.4"
+                print(
+                    "[WARNING] Navi 4x series GPUs require preview ROCm builds (rocm6.4). "
+                    "torchruntime will fall back to CPU unless preview=True is enabled."
+                )
+                return "cpu"
             if any(device_name.startswith("Navi") for device_name in device_names) and any(
                 device_name.startswith("Vega 2") for device_name in device_names
             ):  # lowest-common denominator is rocm5.7, which works with both Navi and Vega 20