Enable CMake options to build dpnp on AMD

Author: vlad-perevezentsev

CMakeLists.txt

@@ -71,31 +71,71 @@ option(DPNP_TARGET_CUDA
     "Build DPNP to target CUDA devices"
     OFF
 )
+option(DPNP_TARGET_HIP
+    "Build DPNP to target HIP devices"
+    OFF
+)
 option(DPNP_USE_ONEMKL_INTERFACES
     "Build DPNP with oneMKL Interfaces"
     OFF
 )
+set(HIP_TARGETS "" CACHE STRING "HIP architecture for target")
+
+#Add rocm root dir to CMAKE_PREFIX_PATH, usually "/opt/rocm"
+if(DPNP_TARGET_HIP)
+    if(DEFINED ENV{ROCM_PATH})
+        set(ROCM_PATH $ENV{ROCM_PATH})
+    else()
+        set(ROCM_PATH "/opt/rocm")
+    endif()
+    list(APPEND CMAKE_PREFIX_PATH ${ROCM_PATH})
+endif()
+
 set(_dpnp_sycl_targets)
+set(_dpnp_amd_targets)
+set(_use_onemkl_interfaces OFF)
 set(_use_onemkl_interfaces_cuda OFF)
+set(_use_onemkl_interfaces_hip OFF)
+
+set(_dpnp_sycl_target_compile_options)
+set(_dpnp_sycl_target_link_options)
+
 if ("x${DPNP_SYCL_TARGETS}" STREQUAL "x")
-   if(DPNP_TARGET_CUDA)
-      set(_dpnp_sycl_targets "nvptx64-nvidia-cuda,spir64-unknown-unknown")
-      set(_use_onemkl_interfaces_cuda ON)
-   else()
-      if(DEFINED ENV{DPNP_TARGET_CUDA})
-          set(_dpnp_sycl_targets "nvptx64-nvidia-cuda,spir64-unknown-unknown")
-          set(_use_onemkl_interfaces_cuda ON)
-      endif()
-   endif()
+    if(DPNP_TARGET_CUDA)
+        set(_dpnp_sycl_targets "nvptx64-nvidia-cuda,spir64-unknown-unknown")
+        set(_use_onemkl_interfaces_cuda ON)
+    endif()
+    if(DPNP_TARGET_HIP)
+        if (NOT "x${HIP_TARGETS}" STREQUAL "x")
+            set(_dpnp_amd_targets ${HIP_TARGETS})
+            set(_use_onemkl_interfaces_hip ON)
+            if(_dpnp_sycl_targets)
+                set(_dpnp_sycl_targets "amdgcn-amd-amdhsa,${_dpnp_sycl_targets}")
+            else()
+                set(_dpnp_sycl_targets "amdgcn-amd-amdhsa,spir64-unknown-unknown")
+            endif()
+        else()
+            message(FATAL_ERROR "HIP_TARGETS must be specified when using HIP backend")
+        endif()
+    endif()
 else()
-   set(_dpnp_sycl_targets ${DPNP_SYCL_TARGETS})
+    set(_dpnp_sycl_targets ${DPNP_SYCL_TARGETS})
+    if (NOT "x${HIP_TARGETS}" STREQUAL "x")
+        set(_dpnp_amd_targets ${HIP_TARGETS})
+        set(_use_onemkl_interfaces_hip ON)
+    endif()
 endif()
 
-if(_dpnp_sycl_targets)
+if (_dpnp_sycl_targets)
    message(STATUS "Compiling for -fsycl-targets=${_dpnp_sycl_targets}")
+   list(APPEND _dpnp_sycl_target_compile_options -fsycl-targets=${_dpnp_sycl_targets})
+   list(APPEND _dpnp_sycl_target_link_options -fsycl-targets=${_dpnp_sycl_targets})
+   if(_dpnp_amd_targets)
+      list(APPEND _dpnp_sycl_target_compile_options -Xsycl-target-backend=amdgcn-amd-amdhsa --offload-arch=${_dpnp_amd_targets})
+      list(APPEND _dpnp_sycl_target_link_options -Xsycl-target-backend=amdgcn-amd-amdhsa --offload-arch=${_dpnp_amd_targets})
+   endif()
 endif()
 
-set(_use_onemkl_interfaces OFF)
 if(DPNP_USE_ONEMKL_INTERFACES)
     set(_use_onemkl_interfaces ON)
 else()
@@ -107,13 +147,20 @@ endif()
 if(_use_onemkl_interfaces)
     set(BUILD_FUNCTIONAL_TESTS False)
     set(BUILD_EXAMPLES False)
+    set(ENABLE_MKLGPU_BACKEND True)
+    set(ENABLE_MKLCPU_BACKEND True)
+
     if(_use_onemkl_interfaces_cuda)
         set(ENABLE_CUBLAS_BACKEND True)
         set(ENABLE_CUSOLVER_BACKEND True)
         set(ENABLE_CUFFT_BACKEND True)
         # set(ENABLE_CURAND_BACKEND True)
-        set(ENABLE_MKLGPU_BACKEND True)
-        set(ENABLE_MKLCPU_BACKEND True)
+    endif()
+    if(_use_onemkl_interfaces_hip)
+        set(ENABLE_ROCBLAS_BACKEND True)
+        set(ENABLE_ROCSOLVER_BACKEND True)
+        set(ENABLE_ROCFFT_BACKEND True)
+        # set(ENABLE_ROCRAND_BACKEND True)
     endif()
 
     if(DPNP_ONEMKL_INTERFACES_DIR)

dpnp/backend/extensions/statistics/CMakeLists.txt

@@ -45,12 +45,14 @@ if(_dpnp_sycl_targets)
     target_compile_options(
         ${python_module_name}
         PRIVATE
-        -fsycl-targets=${_dpnp_sycl_targets}
+        ${_dpnp_sycl_target_compile_options}
+        # -fsycl-targets=${_dpnp_sycl_targets}
     )
     target_link_options(
         ${python_module_name}
         PRIVATE
-        -fsycl-targets=${_dpnp_sycl_targets}
+        ${_dpnp_sycl_target_link_options}
+        # -fsycl-targets=${_dpnp_sycl_targets}
     )
 endif()
 

dpnp/backend/extensions/ufunc/CMakeLists.txt

@@ -77,12 +77,14 @@ if(_dpnp_sycl_targets)
   target_compile_options(
     ${python_module_name}
     PRIVATE
-    -fsycl-targets=${_dpnp_sycl_targets}
+    ${_dpnp_sycl_target_compile_options}
+    # -fsycl-targets=${_dpnp_sycl_targets}
   )
   target_link_options(
     ${python_module_name}
     PRIVATE
-    -fsycl-targets=${_dpnp_sycl_targets}
+    ${_dpnp_sycl_target_link_options}
+    # -fsycl-targets=${_dpnp_sycl_targets}
   )
 endif()