diff options
-rw-r--r-- | unsupported/test/cxx11_tensor_contraction_mkldnn.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp b/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp index f164d644d..8ba0f986d 100644 --- a/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp +++ b/unsupported/test/cxx11_tensor_contraction_mkldnn.cpp @@ -127,5 +127,8 @@ static void test_mkldnn_gemm_kernel() { EIGEN_DECLARE_TEST(cxx11_tensor_contraction_mkldnn) { CALL_SUBTEST(test_mkldnn_gemm_pack<float>()); + CALL_SUBTEST(test_mkldnn_gemm_pack<double>()); + + // mkldnn has only sgemm (aka gemm for floats). CALL_SUBTEST(test_mkldnn_gemm_kernel<float>()); } |