add amgx (#3650)

packages/a/amgx/patches/2.4.0/msvc.patch

@@ -0,0 +1,27 @@
+diff --git a/CMakeLists.txt b/CMakeLists.txt
+index 7bc0f1b9..7b0b606e 100644
+--- a/CMakeLists.txt
++++ b/CMakeLists.txt
+@@ -30,6 +30,7 @@ cmake_minimum_required (VERSION 3.18)
+ project (AMG LANGUAGES C CXX CUDA)
+ 
+ find_package(MPI)
++find_package(OpenMP)
+ 
+ set(CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../cmake" ${CMAKE_MODULE_PATH})
+ 
+diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
+index b7fbef7d..0d77ddc7 100644
+--- a/src/CMakeLists.txt
++++ b/src/CMakeLists.txt
+@@ -107,9 +107,7 @@ ELSE()
+     set(libs_all CUDA::cusparse CUDA::cublas rt dl)
+ ENDIF()
+ 
+-add_dependencies(amgx_tests_launcher amgx_tests_library)
+-
+-target_link_libraries(amgx_tests_launcher amgxsh ${libs_all} OpenMP::OpenMP_C)
++target_link_libraries(amgx_tests_launcher amgx_tests_library amgxsh ${libs_all} OpenMP::OpenMP_C)
+ 
+ if(MPI_FOUND)
+     target_link_libraries(amgx_tests_launcher MPI::MPI_CXX)

packages/a/amgx/xmake.lua

@@ -0,0 +1,42 @@
+package("amgx")
+
+    set_homepage("https://developer.nvidia.com/amgx")
+    set_description("Distributed multigrid linear solver library on GPU")
+    set_license("BSD-3-Clause")
+
+    add_urls("https://github.com/NVIDIA/AMGX.git")
+    add_versions("v2.4.0", "2b4762f02af2ed136134c7f0570646219753ab3e")
+
+    add_patches("2.4.0", "patches/2.4.0/msvc.patch", "46dcb9a5e1b4157fce91e06050c1d70f5e4fe34d7bf085216629c4f8708f90a5")
+
+    if is_plat("windows") then
+        set_policy("platform.longpaths", true)
+        add_configs("shared", {description = "Build shared library.", default = false, type = "boolean", readonly = true})
+    end
+
+    add_deps("cmake", "openmp")
+    on_load("windows", function (package)
+        package:add("deps", "cuda", {system = true, configs = {utils = {"cublas", "cusparse", "cusolver"}}})
+        if not package:config("shared") then
+            package:add("defines", "AMGX_API_NO_IMPORTS")
+        end
+    end)
+
+    on_load("linux", function (package)
+        package:add("deps", "cuda", {system = true, configs = {utils = {"cublas", "cusparse", "cusolver"}}})
+        package:add("deps", "nvtx")
+        package:add("syslinks", "pthread", "m")
+    end)
+
+    on_install("windows", "linux", function (package)
+        io.replace("CMakeLists.txt", "/Zl", "", {plain = true})
+        io.replace("CMakeLists.txt", "add_subdirectory(examples)", "", {plain = true})
+        local configs = {"-DCMAKE_NO_MPI=ON"}
+        table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:is_debug() and "Debug" or "Release"))
+        import("package.tools.cmake").install(package, configs)
+        package:add("links", package:config("shared") and "amgxsh" or "amgx")
+    end)
+
+    on_test(function (package)
+        assert(package:has_cfuncs("AMGX_initialize", {includes = "amgx_c.h"}))
+    end)