From 620b0744912cd0d71f030ef7e9df11687b0477f4 Mon Sep 17 00:00:00 2001 From: Nimish Mishra Date: Mon, 20 May 2024 09:56:30 -0700 Subject: [PATCH] Assign proper backends to CUFFT and ROCFFT --- examples/dft/run_time_dispatching/CMakeLists.txt | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/examples/dft/run_time_dispatching/CMakeLists.txt b/examples/dft/run_time_dispatching/CMakeLists.txt index 9b8deb25f..5f0f97446 100644 --- a/examples/dft/run_time_dispatching/CMakeLists.txt +++ b/examples/dft/run_time_dispatching/CMakeLists.txt @@ -29,8 +29,17 @@ set(DFT_RT_SOURCES "") set(DEVICE_FILTERS "") if(ENABLE_MKLGPU_BACKEND OR ENABLE_CUFFT_BACKEND OR ENABLE_ROCFFT_BACKEND OR ENABLE_PORTFFT_BACKEND) list(APPEND DFT_RT_SOURCES "real_fwd_usm") +endif() + +if(ENABLE_MKLGPU_BACKEND OR ENABLE_PORTFFT_BACKEND) list(APPEND DEVICE_FILTERS "level_zero:gpu") endif() +if(ENABLE_CUFFT_BACKEND) + list(APPEND DEVICE_FILTERS "cuda:gpu") +endif() +if(ENABLE_ROCFFT_BACKEND) + list(APPEND DEVICE_FILTERS "hip:gpu") +endif() message(STATUS "ONEAPI_DEVICE_SELECTOR will be set to the following value(s): [${DEVICE_FILTERS}] for run-time dispatching examples")