Skip to content

Commit 7d491e8

Browse files
Enable build Intel backend in onemkl interfaces on CUDA (#2229)
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.
1 parent d83ea3d commit 7d491e8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,8 +112,8 @@ if(_use_onemkl_interfaces)
112112
set(ENABLE_CUSOLVER_BACKEND True)
113113
set(ENABLE_CUFFT_BACKEND True)
114114
# set(ENABLE_CURAND_BACKEND True)
115-
set(ENABLE_MKLGPU_BACKEND False)
116-
set(ENABLE_MKLCPU_BACKEND False)
115+
set(ENABLE_MKLGPU_BACKEND True)
116+
set(ENABLE_MKLCPU_BACKEND True)
117117
endif()
118118

119119
if(DPNP_ONEMKL_INTERFACES_DIR)

0 commit comments

Comments
 (0)