diff --git a/help_function/test_help.py b/help_function/test_help.py index 1ccb5db39..637d12682 100644 --- a/help_function/test_help.py +++ b/help_function/test_help.py @@ -86,8 +86,9 @@ def build_test(): if (test_config.current_test in blas_cases) or (test_config.current_test in fft_cases) or ( test_config.current_test in lapack_cases) or (test_config.current_test in rng_cases) or ( test_config.current_test in oneDNN_related) or (test_config.current_test in sparse_cases): - mkl_opts = [] - if platform.system() == "Linux": + if test_config.device_filter == "cuda:gpu" : + mkl_opts = test_config.mkl_cuda_link_lin + elif platform.system() == "Linux": mkl_opts = test_config.mkl_link_opt_lin else: mkl_opts = test_config.mkl_link_opt_win diff --git a/test_config.py b/test_config.py index 394b19dec..83a3b0a2d 100644 --- a/test_config.py +++ b/test_config.py @@ -44,6 +44,8 @@ mkl_link_opt_lin = ["-lmkl_intel_ilp64", "-lmkl_sequential", "-lmkl_core", "-lOpenCL", "-lmkl_sycl", "-lpthread", "-ldl"] +mkl_cuda_link_lin = ["-lonemkl"] + mkl_link_opt_win = ["mkl_sycl_dll.lib", "mkl_intel_ilp64_dll.lib", "mkl_sequential_dll.lib", "mkl_core_dll.lib", "OpenCL.lib"] mkl_comp_opt = ["-DMKL_ILP64"]