Hoildkv
4 years ago
committed by
GitHub
parent
6075e90bf7
commit
f3cb77c768
2 changed files with 165 additions and 0 deletions
@ -0,0 +1,77 @@ |
||||
import("lib.detect.find_path") |
||||
import("lib.detect.find_library") |
||||
|
||||
function _find_package(package, opt) |
||||
local rdir = (package:is_arch("x64", "x86_64") and "intel64" or "ia32") |
||||
local paths = { |
||||
"$(env MKL_ROOT)", |
||||
"$(env ONEAPI_ROOT)\\mkl\\latest" |
||||
} |
||||
|
||||
-- find library |
||||
local result = {links = {}, linkdirs = {}, includedirs = {}} |
||||
local linkinfo = find_library("mkl_core", paths, {suffixes = {"lib", path.join("lib", rdir), path.join("lib", rdir, "gcc*")}}) |
||||
if not linkinfo then |
||||
return |
||||
end |
||||
table.insert(result.linkdirs, linkinfo.linkdir) |
||||
if rdir == "intel64" then |
||||
table.insert(result.links, "mkl_intel_ilp64") |
||||
elseif package:is_plat("windows") then |
||||
table.insert(result.links, "mkl_intel_c") |
||||
else |
||||
table.insert(result.links, "mkl_intel") |
||||
end |
||||
|
||||
local threading = package:config("threading") |
||||
if threading then |
||||
if threading == "tbb" then |
||||
table.join2(result.links, {"mkl_tbb_thread", "mkl_core"}) |
||||
package:add("deps", "tbb") |
||||
elseif threading == "seq" then |
||||
table.join2(result.links, {"mkl_sequential", "mkl_core"}) |
||||
elseif threading == "openmp" then |
||||
table.join2(result.links, {"mkl_intel_thread", "mkl_core"}) |
||||
end |
||||
else |
||||
if find_package("tbb") then |
||||
table.join2(result.links, {"mkl_tbb_thread", "mkl_core"}) |
||||
package:add("deps", "tbb") |
||||
else |
||||
table.join2(result.links, {"mkl_sequential", "mkl_core"}) |
||||
end |
||||
end |
||||
|
||||
-- find include |
||||
local includepath = find_path(path.join("mkl.h"), paths, {suffixes = "include"}) |
||||
if includepath then |
||||
table.insert(result.includedirs, includepath) |
||||
end |
||||
|
||||
if #result.includedirs > 0 and #result.linkdirs > 0 then |
||||
local version_file = path.join(includepath, "mkl_version.h") |
||||
if os.isfile(version_file) then |
||||
local content = io.readfile(version_file) |
||||
local major = content:match("__INTEL_MKL__ +(%d+)\n") |
||||
local minor = content:match("__INTEL_MKL_MINOR__ +(%d+)\n") |
||||
local patch = content:match("__INTEL_MKL_UPDATE__ +(%d+)\n") |
||||
if patch then |
||||
result.version = format("%s.%s.%s", major, minor, patch) |
||||
else |
||||
result.version = format("%s.%s", major, minor) |
||||
end |
||||
end |
||||
|
||||
return result |
||||
end |
||||
end |
||||
|
||||
function main(package, opt) |
||||
if opt.system and package.find_package then |
||||
local result = _find_package(package, opt) |
||||
if not result then |
||||
result = package:find_package("mkl", opt) |
||||
end |
||||
return result or false |
||||
end |
||||
end |
||||
@ -0,0 +1,88 @@ |
||||
package("mkl") |
||||
|
||||
set_homepage("https://software.intel.com/content/www/us/en/develop/tools/oneapi/components/onemkl.html") |
||||
set_description("Intel® oneAPI Math Kernel Library") |
||||
|
||||
if is_plat("windows") then |
||||
if is_arch("x64") then |
||||
add_urls("https://anaconda.org/intel/mkl-static/$(version)/download/win-64/mkl-static-$(version)-intel_296.tar.bz2") |
||||
add_versions("2021.2.0", "54209e5d9c4778381f08b9a90e900c001494db020cda426441cd624cb0f7ebdc") |
||||
add_resources("2021.2.0", "headers", "https://anaconda.org/intel/mkl-include/2021.2.0/download/win-64/mkl-include-2021.2.0-intel_296.tar.bz2", "ba222ea4ceb9e09976f23a3df39176148b4469b297275f3d05c1ad411b3d54c3") |
||||
elseif is_arch("x86") then |
||||
add_urls("https://anaconda.org/intel/mkl-static/$(version)/download/win-32/mkl-static-$(version)-intel_296.tar.bz2") |
||||
add_versions("2021.2.0", "eaf0df027d58c5fd948f86b83dfc4d608855962cbdb04551712c9aeeb7b26eca") |
||||
add_resources("2021.2.0", "headers", "https://anaconda.org/intel/mkl-include/2021.2.0/download/win-32/mkl-include-2021.2.0-intel_296.tar.bz2", "8ed173edff75783426de1bbc1d122266047fc84d4cfc5a9b810b1f2792f02c37") |
||||
end |
||||
elseif is_plat("macosx") and is_arch("x86_64") then |
||||
add_urls("https://anaconda.org/intel/mkl-static/$(version)/download/osx-64/mkl-static-$(version)-intel_269.tar.bz2") |
||||
add_versions("2021.2.0", "b7af248f01799873333cbd388b5efa19601cf6815dc38713509974783f4b1ccd") |
||||
add_resources("2021.2.0", "headers", "https://anaconda.org/intel/mkl-include/2021.2.0/download/osx-64/mkl-include-2021.2.0-intel_269.tar.bz2", "5215d62cadeb3f8021230163dc35ad38259e3688aa0f39d7da69ebe54ab45624") |
||||
elseif is_plat("linux") then |
||||
if is_arch("x86_64") then |
||||
add_urls("https://anaconda.org/intel/mkl-static/$(version)/download/linux-64/mkl-static-$(version)-intel_296.tar.bz2") |
||||
add_versions("2021.2.0", "2bcaefefd593e4fb521e1fc88715f672ae5b9d1706babf10e3a10ef43ea0f983") |
||||
add_resources("2021.2.0", "headers", "https://anaconda.org/intel/mkl-include/2021.2.0/download/linux-64/mkl-include-2021.2.0-intel_296.tar.bz2", "13721fead8a3eddee15b914fd3ae9cf2095966af79bbc2f086462eda9fff4d62") |
||||
elseif is_arch("x86") then |
||||
add_urls("https://anaconda.org/intel/mkl-static/$(version)/download/linux-32/mkl-static-$(version)-intel_296.tar.bz2") |
||||
add_versions("2021.2.0", "34a1bc80a4a39ca5a55d29e9fcc803380fbc4d029ae496e60a918e8d12db68c2") |
||||
add_resources("2021.2.0", "headers", "https://anaconda.org/intel/mkl-include/2021.2.0/download/linux-32/mkl-include-2021.2.0-intel_296.tar.bz2", "7fcbc945377b486b40d29b170d0b6c39bbc5b430ac7284dae2046bbf610f643d") |
||||
end |
||||
end |
||||
|
||||
add_configs("threading", {description = "Choose threading modal for mkl.", default = nil, type = "string", values = {"tbb", "openmp", "seq"}}) |
||||
|
||||
on_fetch("fetch") |
||||
|
||||
if is_plat("linux") then |
||||
add_syslinks("pthread", "dl") |
||||
end |
||||
on_load("windows", "macosx", "linux", function (package) |
||||
if package:is_plat("windows") then |
||||
package:add("links", package:is_arch("x64", "x86_64") and "mkl_intel_ilp64" or "mkl_intel_c") |
||||
else |
||||
package:add("links", package:is_arch("x64", "x86_64") and "mkl_intel_ilp64" or "mkl_intel") |
||||
end |
||||
|
||||
local threading = package:config("threading") |
||||
if threading then |
||||
if threading == "tbb" then |
||||
package:add("links", "mkl_tbb_thread") |
||||
package:add("deps", "tbb") |
||||
elseif threading == "seq" then |
||||
package:add("links", "mkl_sequential") |
||||
elseif threading == "openmp" then |
||||
package:add("links", "mkl_intel_thread") |
||||
end |
||||
else |
||||
if find_package("tbb") then |
||||
package:add("links", "mkl_tbb_thread") |
||||
package:add("deps", "tbb") |
||||
else |
||||
package:add("links", "mkl_sequential") |
||||
end |
||||
end |
||||
|
||||
package:add("links", "mkl_core") |
||||
end) |
||||
|
||||
on_install("windows", "macosx", "linux", function (package) |
||||
local headerdir = package:resourcedir("headers") |
||||
if package:is_plat("windows") then |
||||
os.trymv(path.join("Library", "lib"), package:installdir()) |
||||
os.trymv(path.join(headerdir, "Library", "include"), package:installdir()) |
||||
else |
||||
os.trymv(path.join("lib"), package:installdir()) |
||||
os.trymv(path.join(headerdir, "include"), package:installdir()) |
||||
end |
||||
end) |
||||
|
||||
on_test(function (package) |
||||
assert(package:check_csnippets({test = [[ |
||||
void test() { |
||||
double A[6] = {1.0,2.0,1.0,-3.0,4.0,-1.0}; |
||||
double B[6] = {1.0,2.0,1.0,-3.0,4.0,-1.0}; |
||||
double C[9] = {.5,.5,.5,.5,.5,.5,.5,.5,.5}; |
||||
cblas_dgemm(CblasColMajor,CblasNoTrans,CblasTrans,3,3,2,1,A,3,B,3,2,C,3); |
||||
} |
||||
]]}, {includes = "mkl_cblas.h"})) |
||||
end) |
||||
Loading…
Reference in new issue