From 22ed98a331d059dfc437f11158693ec759423e9c Mon Sep 17 00:00:00 2001 From: Eugene Zhulenev Date: Wed, 26 Sep 2018 17:57:37 -0700 Subject: [PATCH] Conditionally add mkldnn test --- unsupported/test/CMakeLists.txt | 5 ++++- unsupported/test/cxx11_tensor_contraction_mkldnn.cpp | 10 ---------- 2 files changed, 4 insertions(+), 11 deletions(-) diff --git a/unsupported/test/CMakeLists.txt b/unsupported/test/CMakeLists.txt index 25ae1a8d2..2fd4e1b97 100644 --- a/unsupported/test/CMakeLists.txt +++ b/unsupported/test/CMakeLists.txt @@ -191,6 +191,10 @@ if(EIGEN_TEST_CXX11) ei_add_test_sycl(cxx11_tensor_custom_op_sycl ${STD_CXX_FLAG}) endif(EIGEN_TEST_SYCL) + if (MKLDNN_FOUND) + ei_add_test(cxx11_tensor_contraction_mkldnn) + endif (MKLDNN_FOUND) + ei_add_test(cxx11_eventcount "-pthread" "${CMAKE_THREAD_LIBS_INIT}") ei_add_test(cxx11_runqueue "-pthread" "${CMAKE_THREAD_LIBS_INIT}") ei_add_test(cxx11_non_blocking_thread_pool "-pthread" "${CMAKE_THREAD_LIBS_INIT}") @@ -201,7 +205,6 @@ if(EIGEN_TEST_CXX11) ei_add_test(cxx11_tensor_index_list) ei_add_test(cxx11_tensor_mixed_indices) ei_add_test(cxx11_tensor_contraction) - ei_add_test(cxx11_tensor_contraction_mkldnn) ei_add_test(cxx11_tensor_convolution) ei_add_test(cxx11_tensor_expr) ei_add_test(cxx11_tensor_fixed_size) diff --git a/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp b/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp index 5a905c0cf..f164d644d 100644 --- a/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp +++ b/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp @@ -9,15 +9,8 @@ #include "main.h" -// Nothing to test here if we do not have mkldnn enabled. -#if defined(EIGEN_USE_MKLDNN) - #include -using Eigen::array; -using Eigen::ColMajor; -using Eigen::Tensor; -using Eigen::Index; using Eigen::internal::blas_data_mapper; using Eigen::internal::mkldnn_gemm_kernel; using Eigen::internal::mkldnn_gemm_pack; @@ -136,6 +129,3 @@ EIGEN_DECLARE_TEST(cxx11_tensor_contraction_mkldnn) { CALL_SUBTEST(test_mkldnn_gemm_pack()); CALL_SUBTEST(test_mkldnn_gemm_kernel()); } -#else -EIGEN_DECLARE_TEST(cxx11_tensor_contraction_mkldnn) {} -#endif // EIGEN_USE_MKLDNN