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 8236103fe66..be747c42a3a 100644 --- a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java +++ b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java @@ -42,9 +42,11 @@ import org.bytedeco.javacpp.annotation.Platform; import org.bytedeco.javacpp.annotation.Properties; +import org.bytedeco.javacpp.tools.BuildEnabled; import org.bytedeco.javacpp.tools.Info; import org.bytedeco.javacpp.tools.InfoMap; import org.bytedeco.javacpp.tools.InfoMapper; +import org.bytedeco.javacpp.tools.Logger; import org.bytedeco.openblas.presets.openblas; @@ -99,7 +101,7 @@ target = "org.bytedeco.pytorch", global = "org.bytedeco.pytorch.global.torch" ) -public class torch implements LoadEnabled, InfoMapper { +public class torch implements LoadEnabled, InfoMapper, BuildEnabled { static { Loader.checkVersion("org.bytedeco", "pytorch"); } @@ -140,8 +142,6 @@ public void init(ClassProperties properties) { initIncludes(getClass(), properties); - arm64 = platform.contains("arm64"); - // Only apply this at load time since we don't want to copy the CUDA libraries here if (!Loader.isLoadLibraries() || extension == null || !extension.endsWith("-gpu")) { return; @@ -191,6 +191,11 @@ public void init(ClassProperties properties) { } } + @Override + public void init(Logger logger, java.util.Properties properties, String encoding) { + arm64 = properties.getProperty("platform").contains("arm64"); + } + public void mapModule(InfoMap infoMap, String name) { mapModule(infoMap, name, null, null, true); }