Fix linking on Windows

bytedeco · Aug 14, 2024 · 197c450 · 197c450
1 parent dbc19eb
commit 197c450
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java
@@ -86,7 +86,7 @@
 
             },
             exclude = {"openblas_config.h", "cblas.h", "lapacke_config.h", "lapacke_mangling.h", "lapack.h", "lapacke.h", "lapacke_utils.h"},
-            link = { "c10", "torch" }
+            link = { "c10", "torch", "torch_cpu" }
         ),
         @Platform(
             value = {"linux", "macosx", "windows"},
@@ -104,26 +104,25 @@
             extension = "-gpu"
         ),
         @Platform(
-            value = {"linux"},
-            preload = { "torch_cpu" }
+            value = {"linux"}
         ),
         @Platform(
             value = {"macosx"},
-            preload = { "torch_cpu", "iomp5" }
+            preload = { "iomp5" }
         ),
         @Platform(
             value = "windows",
-            preload = { "torch_cpu", "uv" }
+            preload = { "uv" }
         ),
         @Platform(
             value = "linux",
             extension = "-gpu",
-            preload = { "torch_cpu", "c10_cuda", "torch_cuda" }
+            preload = { "c10_cuda", "torch_cuda" }
         ),
         @Platform(
             value = "windows",
             extension = "-gpu",
-            preload = { "torch_cpu", "uv", "c10_cuda", "torch_cuda" }
+            preload = { "uv", "c10_cuda", "torch_cuda" }
         )
     },
     target = "org.bytedeco.pytorch",