diff options
Diffstat (limited to 'third_party/mkl_dnn/mkldnn.BUILD')
-rw-r--r-- | third_party/mkl_dnn/mkldnn.BUILD | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/third_party/mkl_dnn/mkldnn.BUILD b/third_party/mkl_dnn/mkldnn.BUILD new file mode 100644 index 0000000000..58bb7a6a5d --- /dev/null +++ b/third_party/mkl_dnn/mkldnn.BUILD @@ -0,0 +1,25 @@ +exports_files(["LICENSE"]) + +cc_library( + name = "mkl_dnn", + srcs = glob([ + "src/common/*.cpp", + "src/cpu/*.cpp", + ]), + hdrs = glob(["include/*"]), + copts = ["-fexceptions"] + select({ + "@org_tensorflow//tensorflow:linux_x86_64": [ + "-fopenmp", + ], + "//conditions:default": [], + }), + includes = [ + "include", + "src", + "src/common", + "src/cpu", + "src/cpu/xbyak", + ], + nocopts = "-fno-exceptions", + visibility = ["//visibility:public"], +) |