Skip to content

Commit

Permalink
Enable build Intel backend in onemkl interfaces on CUDA (#2229)
Browse files Browse the repository at this point in the history
This PR suggests to enable `MKLGPU_BACKEND` and `MKLCPU_BACKEND` builds
in OneMKL Interfaces during build on CUDA with `--target=cuda` flag to
ensure that all available devices can be used

Previous implementation only allowed array allocation on `cuda::gpu`
device with `ONEAPI_DEVICE_SEELCTOR=cuda:gpu` env variable enabled and
threw RuntimeError otherwise.
  • Loading branch information
vlad-perevezentsev authored Dec 12, 2024
1 parent d83ea3d commit 7d491e8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -112,8 +112,8 @@ if(_use_onemkl_interfaces)
set(ENABLE_CUSOLVER_BACKEND True)
set(ENABLE_CUFFT_BACKEND True)
# set(ENABLE_CURAND_BACKEND True)
set(ENABLE_MKLGPU_BACKEND False)
set(ENABLE_MKLCPU_BACKEND False)
set(ENABLE_MKLGPU_BACKEND True)
set(ENABLE_MKLCPU_BACKEND True)
endif()

if(DPNP_ONEMKL_INTERFACES_DIR)
Expand Down

0 comments on commit 7d491e8

Please sign in to comment.