1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- package("clblast")
- set_homepage("https://github.com/CNugteren/CLBlast")
- set_description("Tuned OpenCL BLAS ")
- set_license("Apache-2.0")
- add_urls("https://github.com/CNugteren/CLBlast/archive/$(version).tar.gz",
- "https://github.com/CNugteren/CLBlast.git")
- add_versions("1.6.0", "9bff8219f753262e2c3bb38eb74264dce8772f626ed59d0765851a4269532888")
- add_configs("tuners", { description = "Enable compilation of the tuners", default = false, type = "boolean" })
- add_configs("workaround", { description = "Enables workaround for bug in AMD Southern Island GPUs", default = false, type = "boolean" })
- add_configs("verbose", { description = "Compile in verbose mode for additional diagnostic messages", default = false, type = "boolean" })
- add_configs("opencl", { description = "Build CLBlast with an OpenCL API", default = true, type = "boolean" })
- add_configs("cuda", { description = "Build CLBlast with a CUDA API (beta)", default = false, type = "boolean" })
- add_configs("netlib", { description = "Enable compilation of the CBLAS Netlib API", default = false, type = "boolean" })
- add_configs("netlib_persistent_opencl", { description = "Makes OpenCL device and context in the CBLAS Netlib API static", default = false, type = "boolean" })
- add_deps("cmake")
- if is_plat("linux") then
- add_extsources("apt::libclblast-dev")
- elseif is_plat("macosx") then
- add_extsources("brew::clblast")
- end
- on_load(function (package)
- if package:config("cuda") then
- package:add("deps", "cuda")
- end
- if package:config("opencl") then
- package:add("deps", "opencl")
- end
- end)
- on_install("windows|x86", "windows|x64", "linux", "macosx", function (package)
- local configs = {}
- table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:debug() and "Debug" or "Release"))
- table.insert(configs, "-DBUILD_SHARED_LIBS=" .. (package:config("shared") and "ON" or "OFF"))
- table.insert(configs, "-DTUNERS=" .. (package:config("tuners") and "ON" or "OFF"))
- table.insert(configs, "-DVERBOSE=" .. (package:config("verbose") and "ON" or "OFF"))
- table.insert(configs, "-DAMD_SI_EMPTY_KERNEL_WORKAROUND=" .. (package:config("workaround") and "ON" or "OFF"))
- table.insert(configs, "-DOPENCL=" .. (package:config("opencl") and "ON" or "OFF"))
- table.insert(configs, "-DCUDA=" .. (package:config("cuda") and "ON" or "OFF"))
- table.insert(configs, "-DNETLIB=" .. (package:config("cuda") and "ON" or "OFF"))
- table.insert(configs, "-DNETLIB_PERSISTENT_OPENCL=" .. (package:config("cuda") and "ON" or "OFF"))
- if package:is_plat("windows") and package:config("vs_runtime"):startswith("MT") then
- table.insert(configs,"-DOVERRIDE_MSVC_FLAGS_TO_MT=ON")
- end
- import("package.tools.cmake").install(package, configs)
- end)
- on_test(function (package)
- assert(package:check_cxxsnippets({test = [[
- #include <clblast.h>
- void test () {
- clblast::ClearCache();
- }
- ]]}, {configs = {languages = "c++11"}}))
- assert(package:check_csnippets({test = [[
- #include <clblast_c.h>
- void test () {
- CLBlastClearCache();
- }
- ]]}, {configs = {languages = "c11"}}))
- end)
|