[UCX] add rocm support (#2489)

JuliaPackaging · Feb 3, 2021 · b179f72 · b179f72 · giordano · Feb 3, 2021
1 parent 1db8477
commit b179f72
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/U/UCX/build_tarballs.jl b/U/UCX/build_tarballs.jl
@@ -45,6 +45,10 @@ if [[ "${target}" != *aarch64* ]]; then
     FLAGS+=(--with-cuda=${prefix}/cuda)
 fi
 
+if [[ "${target}" == *x86_64* ]]; then
+    FLAGS+=(--with-rocm=${prefix})
+fi
+
 ./configure ${FLAGS[@]}
 
 # For a bug in `src/uct/sm/cma/Makefile` that I did't have the time to look
@@ -83,12 +87,14 @@ products = [
 # - ROCM -> TODO
 
 cuda_version = v"11.2.0"
+rocm_version = v"3.7.0"
 
 dependencies = [
     Dependency(PackageSpec(name="CompilerSupportLibraries_jll", uuid="e66e0078-7015-5450-92f7-15fbd957f2ae")),
     Dependency(PackageSpec(name="NUMA_jll", uuid="7f51dc2b-bb24-59f8-b771-bb1490e4195d")),
     Dependency(PackageSpec(name="rdma_core_jll", uuid="69dc3629-5c98-505f-8bcd-225213cebe70")),
-    BuildDependency(PackageSpec(name="CUDA_full_jll", version=cuda_version))
+    BuildDependency(PackageSpec(name="CUDA_full_jll", version=cuda_version)),
+    BuildDependency(PackageSpec(name="hsa_rocr_jll", version=rocm_version))
 ]
 
 # Build the tarballs, and possibly a `build.jl` as well.