From 852b447bab045fb522c2205fa736c9eda07107a8 Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Tue, 30 Jan 2024 23:08:07 +0100
Subject: [PATCH] Fix the build names.

---
 torch-sys/build.rs | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/torch-sys/build.rs b/torch-sys/build.rs
index 0130ffef..c66087ca 100644
--- a/torch-sys/build.rs
+++ b/torch-sys/build.rs
@@ -312,11 +312,8 @@ impl SystemInfo {
                     "https://download.pytorch.org/libtorch/{}/libtorch-cxx11-abi-shared-with-deps-{}{}.zip",
                     device, TORCH_VERSION, match device.as_ref() {
                         "cpu" => "%2Bcpu",
-                        "cu102" => "%2Bcu102",
-                        "cu113" => "%2Bcu113",
-                        "cu116" => "%2Bcu116",
-                        "cu117" => "%2Bcu117",
                         "cu118" => "%2Bcu118",
+                        "cu121" => "%2Bcu121",
                         _ => anyhow::bail!("unsupported device {device}, TORCH_CUDA_VERSION may be set incorrectly?"),
                     }
                 ),
@@ -331,18 +328,15 @@ impl SystemInfo {
                             export DYLD_LIBRARY_PATH=${{LIBTORCH}}/lib
                             ")
                     } else {
-                        format!("https://download.pytorch.org/libtorch/cpu/libtorch-macos-{TORCH_VERSION}.zip")
+                        format!("https://download.pytorch.org/libtorch/cpu/libtorch-macos-x86_64-{TORCH_VERSION}.zip")
                     }
                 },
                 Os::Windows => format!(
                     "https://download.pytorch.org/libtorch/{}/libtorch-win-shared-with-deps-{}{}.zip",
                     device, TORCH_VERSION, match device.as_ref() {
                         "cpu" => "%2Bcpu",
-                        "cu102" => "%2Bcu102",
-                        "cu113" => "%2Bcu113",
-                        "cu116" => "%2Bcu116",
-                        "cu117" => "%2Bcu117",
                         "cu118" => "%2Bcu118",
+                        "cu121" => "%2Bcu121",
                         _ => ""
                     }),
             };