add mkl (#407)

* add mkl

* fix mkl

packages/m/mkl/fetch.lua

@@ -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

packages/m/mkl/xmake.lua

@@ -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)