xmake.lua 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. package("kokkos")
  2. set_homepage("https://kokkos.github.io/")
  3. set_description("Kokkos C++ Performance Portability Programming EcoSystem: The Programming Model")
  4. set_license("Apache-2.0")
  5. add_urls("https://github.com/kokkos/kokkos/archive/refs/tags/$(version).tar.gz",
  6. "https://github.com/kokkos/kokkos.git")
  7. add_versions("4.4.00", "c638980cb62c34969b8c85b73e68327a2cb64f763dd33e5241f5fd437170205a")
  8. add_versions("4.3.01", "5998b7c732664d6b5e219ccc445cd3077f0e3968b4be480c29cd194b4f45ec70")
  9. add_versions("4.3.00", "53cf30d3b44dade51d48efefdaee7a6cf109a091b702a443a2eda63992e5fe0d")
  10. add_versions("4.2.01", "cbabbabba021d00923fb357d2e1b905dda3838bd03c885a6752062fe03c67964")
  11. add_versions("4.2.00", "ac08765848a0a6ac584a0a46cd12803f66dd2a2c2db99bb17c06ffc589bf5be8")
  12. add_versions("4.0.01", "bb942de8afdd519fd6d5d3974706bfc22b6585a62dd565c12e53bdb82cd154f0")
  13. if is_plat("windows") then
  14. add_configs("shared", {description = "Build shared library.", default = false, type = "boolean", readonly = true})
  15. end
  16. add_configs("threads", {description = "Enable thread support.", default = true, type = "boolean"})
  17. add_configs("cuda", {description = "Enable CUDA support.", default = false, type = "boolean"})
  18. add_configs("arch", {description = "Enable architecture-specific optimizations.", default = (is_plat("macosx", "linux") and "native" or nil), type = "string"})
  19. add_deps("cmake")
  20. add_links("kokkoscontainers", "kokkossimd", "kokkoscore")
  21. on_load("windows|x64", "macosx", "linux", function (package)
  22. if package:config("cuda") then
  23. package:add("deps", "cuda")
  24. end
  25. end)
  26. on_install("windows|x64", "macosx|x86_64", "linux", function (package)
  27. local configs = {"-DKokkos_ENABLE_SERIAL=ON"}
  28. table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:debug() and "Debug" or "Release"))
  29. table.insert(configs, "-DBUILD_SHARED_LIBS=" .. (package:config("shared") and "ON" or "OFF"))
  30. table.insert(configs, "-DKokkos_ENABLE_THREADS=" .. (package:config("threads") and "ON" or "OFF"))
  31. table.insert(configs, "-DKokkos_ENABLE_CUDA=" .. (package:config("cuda") and "ON" or "OFF"))
  32. if package:config("arch") ~= nil then
  33. table.insert(configs, "-DKokkos_ARCH_" .. package:config("arch"):upper() .. "=ON")
  34. end
  35. import("package.tools.cmake").install(package, configs)
  36. end)
  37. on_test(function (package)
  38. assert(package:check_cxxsnippets({test = [[
  39. void test(int argc, char *argv[]) {
  40. Kokkos::initialize(argc, argv);
  41. Kokkos::finalize();
  42. }
  43. ]]}, {configs = {languages = "c++17"}, includes = "Kokkos_Core.hpp"}))
  44. end)