Fix linking

bytedeco · Aug 14, 2024 · dbc19eb · dbc19eb
1 parent da86531
commit dbc19eb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 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 = { "torch" }
+            link = { "c10", "torch" }
         ),
         @Platform(
             value = {"linux", "macosx", "windows"},
@@ -105,25 +105,25 @@
         ),
         @Platform(
             value = {"linux"},
-            preload = { "c10", "torch_cpu" }
+            preload = { "torch_cpu" }
         ),
         @Platform(
             value = {"macosx"},
-            preload = { "c10", "torch_cpu", "iomp5" }
+            preload = { "torch_cpu", "iomp5" }
         ),
         @Platform(
             value = "windows",
-            preload = { "c10", "torch_cpu", "uv" }
+            preload = { "torch_cpu", "uv" }
         ),
         @Platform(
             value = "linux",
             extension = "-gpu",
-            preload = { "c10", "torch_cpu", "c10_cuda", "torch_cuda" }
+            preload = { "torch_cpu", "c10_cuda", "torch_cuda" }
         ),
         @Platform(
             value = "windows",
             extension = "-gpu",
-            preload = { "c10", "torch_cpu", "uv", "c10_cuda", "torch_cuda" }
+            preload = { "torch_cpu", "uv", "c10_cuda", "torch_cuda" }
         )
     },
     target = "org.bytedeco.pytorch",

diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch_cuda.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch_cuda.java
@@ -58,12 +58,12 @@
         @Platform(
             value = "linux",
             extension = "-gpu",
-            link = { "torch" , "c10_cuda", "torch_cuda_linalg" } // cuda_linalg built as separate lib on linux only
+            link = { "c10", "torch" , "c10_cuda", "torch_cuda_linalg" } // cuda_linalg built as separate lib on linux only
         ),
         @Platform(
             value = "windows",
             extension = "-gpu",
-            link = { "torch" , "c10_cuda" }
+            link = { "c10", "torch" , "c10_cuda" }
         )
     },
     target = "org.bytedeco.pytorch.cuda",