Use regular torch wheels index for xpu instead of nightly

Author: cmdr2

tests/test_installer.py

@@ -72,25 +72,23 @@ def test_xpu_platform_windows_with_torch_only(monkeypatch):
     monkeypatch.setattr("torchruntime.installer.os_name", "Windows")
     packages = ["torch"]
     result = get_install_commands("xpu", packages)
-    expected_url = "https://download.pytorch.org/whl/test/xpu"
+    expected_url = "https://download.pytorch.org/whl/xpu"
     assert result == [packages + ["--index-url", expected_url]]
 
 
-def test_xpu_platform_windows_with_torchvision(monkeypatch, capsys):
+def test_xpu_platform_windows_with_torchvision(monkeypatch):
     monkeypatch.setattr("torchruntime.installer.os_name", "Windows")
     packages = ["torch", "torchvision"]
     result = get_install_commands("xpu", packages)
-    expected_url = "https://download.pytorch.org/whl/nightly/xpu"
+    expected_url = "https://download.pytorch.org/whl/xpu"
     assert result == [packages + ["--index-url", expected_url]]
-    captured = capsys.readouterr()
-    assert "[WARNING]" in captured.out
 
 
 def test_xpu_platform_linux(monkeypatch):
     monkeypatch.setattr("torchruntime.installer.os_name", "Linux")
     packages = ["torch", "torchvision"]
     result = get_install_commands("xpu", packages)
-    expected_url = "https://download.pytorch.org/whl/test/xpu"
+    expected_url = "https://download.pytorch.org/whl/xpu"
     triton_index_url = "https://download.pytorch.org/whl"
     assert result == [
         packages + ["--index-url", expected_url],

torchruntime/installer.py

@@ -41,7 +41,6 @@ def get_install_commands(torch_platform, packages):
         ValueError: If an unsupported platform is provided.
 
     Notes:
-        - For "xpu" on Windows, if torchvision or torchaudio are included, the function switches to nightly builds.
         - For "directml", the "torch-directml" package is returned as part of the installation commands.
         - For "ipex", the "intel-extension-for-pytorch" package is returned as part of the installation commands.
         - For Windows CUDA, the function also installs "triton-windows" (for torch.compile and Triton kernels).
@@ -69,15 +68,7 @@ def get_install_commands(torch_platform, packages):
         return cmds
 
     if torch_platform == "xpu":
-        if os_name == "Windows" and ("torchvision" in packages or "torchaudio" in packages):
-            print(
-                f"[WARNING] The preview build of 'xpu' on Windows currently only supports torch, not torchvision/torchaudio. "
-                f"torchruntime will instead use the nightly build, to get the 'xpu' version of torchaudio and torchvision as well. "
-                f"Please contact torchruntime if this is no longer accurate: {CONTACT_LINK}"
-            )
-            index_url = f"https://download.pytorch.org/whl/nightly/{torch_platform}"
-        else:
-            index_url = f"https://download.pytorch.org/whl/test/{torch_platform}"
+        index_url = f"https://download.pytorch.org/whl/{torch_platform}"
 
         cmds = [packages + ["--index-url", index_url]]
         if os_name == "Linux":