diff options
Diffstat (limited to 'third_party')
-rw-r--r-- | third_party/mkl_dnn/mkldnn.BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/third_party/mkl_dnn/mkldnn.BUILD b/third_party/mkl_dnn/mkldnn.BUILD index 57d2e1292b..cd1f9a6327 100644 --- a/third_party/mkl_dnn/mkldnn.BUILD +++ b/third_party/mkl_dnn/mkldnn.BUILD @@ -18,6 +18,7 @@ cc_library( srcs = glob([ "src/common/*.cpp", "src/cpu/*.cpp", + "src/cpu/gemm/*.cpp" ]), hdrs = glob(["include/*"]), copts = [ @@ -42,6 +43,7 @@ cc_library( "src/common", "src/cpu", "src/cpu/xbyak", + "src/cpu/gemm" ], nocopts = "-fno-exceptions", visibility = ["//visibility:public"], |