diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2018-09-26 18:22:24 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2018-09-26 18:22:24 -0700 |
| commit | b314376f9c6d69208b437ae59b412aa57aefd2ec (patch) | |
| tree | 1257f19b0e21f6fa8a9fa4cac9f34d252224ed3c | |
| parent | 22ed98a331d059dfc437f11158693ec759423e9c (diff) | |
Test mkldnn pack for doubles
| -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>()); } |