-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy path0001-enable-mkldnn_rnn-in-tensorflow-codebase.patch
39 lines (36 loc) · 1.45 KB
/
0001-enable-mkldnn_rnn-in-tensorflow-codebase.patch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
From 143a284e3c1f4833d2626b451b123a4528389845 Mon Sep 17 00:00:00 2001
From: YAO Matrix <yaoweifeng0301@126.com>
Date: Wed, 19 Jul 2017 15:55:32 +0800
Subject: [PATCH] enable mkldnn_rnn in tensorflow codebase
Signed-off-by: YAO Matrix <yaoweifeng0301@126.com>
---
tensorflow/contrib/BUILD | 1 +
tensorflow/tensorflow.bzl | 2 +-
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD
index 1ac7e53..195244f 100755
--- a/tensorflow/contrib/BUILD
+++ b/tensorflow/contrib/BUILD
@@ -20,6 +20,7 @@ py_library(
"//tensorflow/contrib/copy_graph:copy_graph_py",
"//tensorflow/contrib/crf:crf_py",
"//tensorflow/contrib/cudnn_rnn:cudnn_rnn_py",
+ "//tensorflow/contrib/mkldnn_rnn:mkldnn_rnn_py",
"//tensorflow/contrib/data",
"//tensorflow/contrib/deprecated:deprecated_py",
"//tensorflow/contrib/distributions:distributions_py",
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 348745f..8e0aee7 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -127,7 +127,7 @@ def tf_copts():
"-DEIGEN_AVOID_STL_ARRAY",
"-Iexternal/gemmlowp",
"-Wno-sign-compare",
- "-fno-exceptions",
+ "-fexceptions",
] + if_cuda(["-DGOOGLE_CUDA=1"]) + if_mkl(["-DINTEL_MKL=1"]) + if_android_arm(
["-mfpu=neon"]) + if_x86(["-msse3"]) + select({
clean_dep("//tensorflow:android"): [
--
1.8.3.1