aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/rnn/ops
diff options
context:
space:
mode:
authorGravatar Suharsh Sivakumar <suharshs@google.com>2017-01-24 15:58:55 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-01-24 16:09:44 -0800
commitd9ae30a17a0fea8666db00e13bcfb760be998988 (patch)
treed4f9830288a45b611bd06394f89bdd353ff8f8ca /tensorflow/contrib/rnn/ops
parent5507516cc1aafff5a51472eb9e4571236a3a94fe (diff)
Remove explicit static linking from tests that load a shared library.
This was causing the .so files to have undefined symbols from core/framework. Change: 145479847
Diffstat (limited to 'tensorflow/contrib/rnn/ops')
-rw-r--r--tensorflow/contrib/rnn/ops/gru_ops_test.cc4
-rw-r--r--tensorflow/contrib/rnn/ops/lstm_ops_test.cc6
2 files changed, 7 insertions, 3 deletions
diff --git a/tensorflow/contrib/rnn/ops/gru_ops_test.cc b/tensorflow/contrib/rnn/ops/gru_ops_test.cc
index de6914fccb..5372e6ba2c 100644
--- a/tensorflow/contrib/rnn/ops/gru_ops_test.cc
+++ b/tensorflow/contrib/rnn/ops/gru_ops_test.cc
@@ -27,8 +27,10 @@ class GruOpsTest : public ::testing::Test {
TF_Status* status = TF_NewStatus();
auto* lib = TF_LoadLibrary(
"tensorflow/contrib/rnn/python/ops/_gru_ops.so", status);
- CHECK_EQ(TF_OK, TF_GetCode(status));
+ TF_Code code = TF_GetCode(status);
+ string status_msg(TF_Message(status));
TF_DeleteStatus(status);
+ ASSERT_EQ(TF_OK, code) << status_msg;
TF_DeleteLibraryHandle(lib);
}
};
diff --git a/tensorflow/contrib/rnn/ops/lstm_ops_test.cc b/tensorflow/contrib/rnn/ops/lstm_ops_test.cc
index a25e0674ea..7c4d003b83 100644
--- a/tensorflow/contrib/rnn/ops/lstm_ops_test.cc
+++ b/tensorflow/contrib/rnn/ops/lstm_ops_test.cc
@@ -29,9 +29,11 @@ class LSTMOpsTest : public ::testing::Test {
TF_Status* status = TF_NewStatus();
auto* lib = TF_LoadLibrary(
"tensorflow/contrib/rnn/python/ops/_lstm_ops.so", status);
- CHECK_EQ(TF_OK, TF_GetCode(status));
- TF_DeleteLibraryHandle(lib);
+ TF_Code code = TF_GetCode(status);
+ string status_msg(TF_Message(status));
TF_DeleteStatus(status);
+ ASSERT_EQ(TF_OK, code) << status_msg;
+ TF_DeleteLibraryHandle(lib);
}
};