From 0d506e65517afc8b4a510855d2d322e8d55dcd55 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20Guillemet?= Date: Fri, 16 Aug 2024 18:50:13 +0200 Subject: [PATCH] Fix linking of torch_cuda --- .../main/java/org/bytedeco/pytorch/presets/torch_cuda.java | 4 ++-- 1 file changed, 2 insertions(+), 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 index c85c457051..fc3271e5b7 100644 --- 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",