update libtorch (#1956)

* update openssl

* update libtorch
pull/2810/head
Hoildkv 1 year ago committed by GitHub
parent 7fbd1b25a0
commit e6867621fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 44
      packages/l/libtorch/xmake.lua
  2. 1
      packages/o/openssl/xmake.lua

@ -12,6 +12,7 @@ package("libtorch")
add_versions("v1.9.1", "dfbd030854359207cb3040b864614affeace11ce")
add_versions("v1.11.0", "bc2c6edaf163b1a1330e37a6e34caf8c553e4755")
add_versions("v1.12.1", "664058fa83f1d8eede5d66418abff6e20bd76ca8")
add_versions("v2.1.0", "7bcf7da3a268b435777fe87c7794c382f444e86d")
add_patches("1.9.x", path.join(os.scriptdir(), "patches", "1.9.0", "gcc11.patch"), "4191bb3296f18f040c230d7c5364fb160871962d6278e4ae0f8bc481f27d8e4b")
add_patches("1.11.0", path.join(os.scriptdir(), "patches", "1.11.0", "gcc11.patch"), "1404b0bc6ce7433ecdc59d3412e3d9ed507bb5fd2cd59134a254d7d4a8d73012")
@ -72,9 +73,10 @@ package("libtorch")
on_install("windows|x64", "macosx", "linux", function (package)
import("package.tools.cmake")
import("core.tool.toolchain")
if package:is_plat("windows") then
local vs = import("core.tool.toolchain").load("msvc"):config("vs")
local vs = toolchain.load("msvc"):config("vs")
if tonumber(vs) < 2019 then
raise("Your compiler is too old to use this library.")
end
@ -113,22 +115,26 @@ package("libtorch")
-- some patches to the third-party cmake files
io.replace("third_party/fbgemm/CMakeLists.txt", "PRIVATE FBGEMM_STATIC", "PUBLIC FBGEMM_STATIC", {plain = true})
-- Workaround to compile with GCC-12.
-- Refer to [this pytorch issue](https://github.com/pytorch/pytorch/issues/77939).
io.replace("third_party/fbgemm/CMakeLists.txt",
'string(APPEND CMAKE_CXX_FLAGS " -Werror")',
'string(APPEND CMAKE_CXX_FLAGS " -Werror")\n string(APPEND CMAKE_CXX_FLAGS " -Wno-uninitialized")',
{plain = true}
)
io.replace("third_party/protobuf/cmake/install.cmake", "install%(DIRECTORY.-%)", "")
if package:is_plat("windows") and package:config("vs_runtime"):startswith("MD") then
io.replace("third_party/fbgemm/CMakeLists.txt", "MT", "MD", {plain = true})
if package:is_plat("windows") then
if package:config("vs_runtime"):startswith("MD") then
io.replace("third_party/fbgemm/CMakeLists.txt", "MT", "MD", {plain = true})
io.replace("c10/macros/Macros.h", "extern \"C\" {\nC10_IMPORT", "extern \"C\" {\n__declspec(dllimport)", {plain = true})
else
io.replace("CMakeLists.txt", "\"NOT BUILD_SHARED_LIBS\" OFF", "\"NOT BUILD_SHARED_LIBS\" ON", {plain = true})
io.replace("c10/macros/Macros.h", "extern \"C\" {\nC10_IMPORT", "extern \"C\" {", {plain = true})
end
end
-- prepare python
os.vrun("python -m pip install typing_extensions pyyaml")
if package:is_plat("windows") then
os.vrun("python -m pip install typing_extensions pyyaml")
else
os.vrun("python3 -m pip install typing_extensions pyyaml")
end
local configs = {"-DUSE_MPI=OFF",
"-DCMAKE_INSTALL_LIBDIR=lib",
"-DUSE_NUMA=OFF",
"-DUSE_MAGMA=OFF",
"-DBUILD_TEST=OFF",
"-DATEN_NO_TEST=ON"}
if package:config("python") then
@ -140,7 +146,7 @@ package("libtorch")
end
-- prepare for installation
local envs = cmake.buildenvs(package, {cmake_generator = "Ninja"})
local envs = cmake.buildenvs(package)
if not package:is_plat("macosx") then
if package:config("blas") == "mkl" then
table.insert(configs, "-DBLAS=MKL")
@ -163,8 +169,18 @@ package("libtorch")
table.insert(configs, "-DUSE_DISTRIBUTED=" .. (package:config("distributed") and "ON" or "OFF"))
table.insert(configs, "-DUSE_SYSTEM_PYBIND11=" .. (package:config("pybind11") and "ON" or "OFF"))
table.insert(configs, "-DBUILD_CUSTOM_PROTOBUF=" .. (package:config("protobuf-cpp") and "OFF" or "ON"))
local pythonpath, err = os.iorun("python -c \"import sys; print(sys.executable)\"")
table.insert(configs, "-DPYTHON_EXECUTABLE=" .. pythonpath)
if package:is_plat("windows") then
table.insert(configs, "-DCAFFE2_USE_MSVC_STATIC_RUNTIME=" .. (package:config("vs_runtime"):startswith("MT") and "ON" or "OFF"))
table.insert(configs, "-DCPUINFO_RUNTIME_TYPE=" .. (package:config("vs_runtime"):startswith("MT") and "static" or "shared"))
local vs_sdkver = toolchain.load("msvc"):config("vs_sdkver")
if vs_sdkver then
local build_ver = string.match(vs_sdkver, "%d+%.%d+%.(%d+)%.?%d*")
assert(tonumber(build_ver) >= 18362, "libtorch requires Windows SDK to be at least 10.0.18362.0")
table.insert(configs, "-DCMAKE_VS_WINDOWS_TARGET_PLATFORM_VERSION=" .. vs_sdkver)
table.insert(configs, "-DCMAKE_SYSTEM_VERSION=" .. vs_sdkver)
end
end
local opt = {envs = envs}
@ -211,5 +227,5 @@ package("libtorch")
auto b = torch::tensor({1, 2, 3});
auto c = torch::dot(a, b);
}
]]}, {configs = {languages = "c++14"}, includes = "torch/torch.h"}))
]]}, {configs = {languages = "c++17"}, includes = "torch/torch.h"}))
end)

@ -6,6 +6,7 @@ package("openssl")
add_urls("https://github.com/openssl/openssl/archive/refs/tags/OpenSSL_$(version).zip", {version = function (version)
return version:gsub("^(%d+)%.(%d+)%.(%d+)-?(%a*)$", "%1_%2_%3%4")
end, excludes = "*/fuzz/*"})
add_versions("1.1.1-w", "c543d84c06e909bfa862d5256f6d9dad502352a580b1193ae262874e4be4d7f7")
add_versions("1.1.1-t", "9422774f3ce0f9cb4db5d862efbf43b4fcc096b37b4ac7157e7c5172113a8a22")
add_patches("1.1.1-t", path.join(os.scriptdir(), "patches", "1.1.1t.diff"), "8009edde46e5577213212ec68f8a40451ceb514fddd892240b1465213b7d2d11")
add_versions("1.1.1-s", "aa76dc0488bc67f5c964d085e37a0f5f452e45c68967816a336fa00f537f5cc5")

Loading…
Cancel
Save