xmake.lua 3.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. package("onedpl")
  2. set_kind("library", {headeronly = true})
  3. set_homepage("https://www.intel.com/content/www/us/en/developer/tools/oneapi/dpc-library.html")
  4. set_description("oneAPI DPC++ Library")
  5. add_urls("https://github.com/oneapi-src/oneDPL/archive/refs/tags/oneDPL-$(version).tar.gz")
  6. add_urls("https://github.com/oneapi-src/oneDPL/archive/refs/tags/oneDPL-$(version)-release.tar.gz")
  7. add_versions("2021.6.1", "4995fe2ed2724b89cdb52c4b6c9af22e146b48d2561abdafdaaa06262dbd67c4")
  8. add_versions("2022.5.0-rc1", "9180c60331ec5b307dd89a5d8bfcd096667985c6761c52322405d4b69193ed88")
  9. add_configs("backend", {description = "Choose threading backend.", default = "tbb", type = "string", values = {"tbb", "dpcpp", "dpcpp_only", "omp", "serial"}})
  10. add_deps("cmake")
  11. on_fetch("fetch")
  12. on_load("windows", "linux", function (package)
  13. local backend = package:config("backend")
  14. if backend == "tbb" then
  15. package:add("deps", "tbb")
  16. package:add("defines", "ONEDPL_USE_TBB_BACKEND=1")
  17. package:add("ldflags", "-ltbb")
  18. elseif backend == "omp" then
  19. package:add("deps", "openmp")
  20. package:add("defines", "ONEDPL_USE_OPENMP_BACKEND=1")
  21. elseif backend == "dpcpp" then
  22. package:add("deps", "tbb")
  23. package:add("ldflags", "-ltbb")
  24. package:add("defines", "ONEDPL_USE_TBB_BACKEND=1")
  25. package:add("defines", "ONEDPL_USE_DPCPP_BACKEND=1")
  26. elseif backend == "dpcpp_only" then
  27. package:add("defines", "ONEDPL_USE_TBB_BACKEND=0")
  28. package:add("defines", "ONEDPL_USE_DPCPP_BACKEND=1")
  29. elseif backend == "serial" then
  30. package:add("defines", "ONEDPL_USE_OPENMP_BACKEND=0")
  31. package:add("defines", "ONEDPL_USE_TBB_BACKEND=0")
  32. package:add("defines", "ONEDPL_USE_DPCPP_BACKEND=0")
  33. end
  34. if package:is_plat("windows") then
  35. package:add("cxxflags", "/Zc:__cplusplus")
  36. end
  37. end)
  38. on_install("windows", "linux", function (package)
  39. io.replace("CMakeLists.txt", "add_subdirectory(test)", "", {plain = true})
  40. -- c.f. https://github.com/oneapi-src/oneDPL/issues/1602 and https://github.com/oneapi-src/oneDPL/commit/e25afef957b50536c5091ed23150fff10921b18f
  41. io.replace("include/oneapi/dpl/pstl/algorithm_impl.h", "(_PSTL_UDR_PRESENT || _ONEDPL_UDR_PRESENT)", "_ONEDPL_UDR_PRESENT // _PSTL_UDR_PRESENT", {plain = true})
  42. io.replace("include/oneapi/dpl/pstl/numeric_impl.h", "(_PSTL_UDS_PRESENT || _ONEDPL_UDS_PRESENT)", "_ONEDPL_UDS_PRESENT // PSTL_UDS_PRESENT", {plain = true})
  43. local configs = {}
  44. table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:debug() and "Debug" or "Release"))
  45. table.insert(configs, "-DONEDPL_BACKEND=" .. package:config("backend"))
  46. import("package.tools.cmake").install(package, configs)
  47. end)
  48. on_test(function (package)
  49. assert(package:check_cxxsnippets({test = [[
  50. #include <oneapi/dpl/algorithm>
  51. #include <oneapi/dpl/execution>
  52. #include <oneapi/dpl/numeric>
  53. #include <vector>
  54. void test() {
  55. const size_t size = 10000000;
  56. std::vector<double> v1(size), v2(size);
  57. double res = std::transform_reduce(oneapi::dpl::execution::par_unseq,
  58. v1.cbegin(), v1.cend(), v2.cbegin(), .0);
  59. }
  60. ]]}, {configs = {languages = "c++17"}}))
  61. end)