From 197c450ec04a5282621640371be14d709de2a53e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20Guillemet?= Date: Wed, 14 Aug 2024 23:02:00 +0200 Subject: [PATCH] Fix linking on Windows --- .../java/org/bytedeco/pytorch/presets/torch.java | 13 ++++++------- 1 file changed, 6 insertions(+), 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 index f3adfc8873..661667b3de 100644 --- 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",