File tree Expand file tree Collapse file tree 2 files changed +10
-2
lines changed
examples/rng/run_time_dispatching Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -145,6 +145,10 @@ if(WIN32 AND ONEMKL_SYCL_IMPLEMENTATION STREQUAL "dpc++")
145145 set (CMAKE_CXX_COMPILE_OBJECT "<CMAKE_CXX_COMPILER> -fsycl /nologo <DEFINES> <INCLUDES> /EHsc <FLAGS> /Fo<OBJECT> -c <SOURCE>" )
146146 set (CMAKE_CXX_CREATE_STATIC_LIBRARY "lib /nologo <OBJECTS> /out:<TARGET>" )
147147 set (CMAKE_CXX_LINK_EXECUTABLE "<CMAKE_CXX_COMPILER> -fsycl /nologo <OBJECTS> -o <TARGET> <LINK_LIBRARIES>" )
148+ find_library (MKL_SYCL_LIB NAMES mkl_sycl
149+ HINTS $ENV{MKLROOT} ${MKL_ROOT}
150+ PATH_SUFFIXES lib/intel64)
151+ set (CMAKE_CXX_CREATE_SHARED_LIBRARY "<CMAKE_CXX_COMPILER> -fsycl /nologo <OBJECTS> ${MKL_SYCL_LIB} /link /out:<TARGET> /implib:<TARGET_IMPLIB> /pdb:<TARGET_PDB> /dll /version:<TARGET_VERSION_MAJOR>.<TARGET_VERSION_MINOR> <LINK_FLAGS> <LINK_LIBRARIES>" )
148152endif ()
149153
150154# Temporary disable sycl 2020 deprecations warnings for cuBLAS and cuSOLVER
Original file line number Diff line number Diff line change @@ -29,12 +29,16 @@ set(DEVICE_FILTERS "")
2929if (ENABLE_MKLCPU_BACKEND)
3030 list (APPEND DEVICE_FILTERS "cpu" )
3131endif ()
32- if (ENABLE_MKLGPU_BACKEND)
32+ # RNG only supports mklcpu backend on Windows
33+ if (UNIX AND ENABLE_MKLGPU_BACKEND)
3334 list (APPEND DEVICE_FILTERS "gpu" )
3435endif ()
35- if (ENABLE_CURAND_BACKEND)
36+ if (UNIX AND ENABLE_CURAND_BACKEND)
3637 list (APPEND DEVICE_FILTERS "cuda:gpu" )
3738endif ()
39+ if (UNIX AND ENABLE_ROCRAND_BACKEND)
40+ list (APPEND DEVICE_FILTERS "hip:gpu" )
41+ endif ()
3842
3943message (STATUS "SYCL_DEVICE_FILTER will be set to the following value(s): [${DEVICE_FILTERS} ] for run-time dispatching examples" )
4044
You can’t perform that action at this time.
0 commit comments