Fix linking of torch_cuda

bytedeco · Aug 16, 2024 · 0d506e6 · 0d506e6
1 parent c88f4f7
commit 0d506e6
Showing 1 changed file with 2 additions and 2 deletions.
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 = { "c10", "torch" , "c10_cuda", "torch_cuda_linalg" } // cuda_linalg built as separate lib on linux only
+            link = { "c10", "torch" , "c10_cuda", "torch_cuda", "torch_cuda_linalg" } // cuda_linalg built as separate lib on linux only
         ),
         @Platform(
             value = "windows",
             extension = "-gpu",
-            link = { "c10", "torch" , "c10_cuda" }
+            link = { "c10", "torch" , "c10_cuda", "torch_cuda" }
         )
     },
     target = "org.bytedeco.pytorch.cuda",