diff options
author | Suharsh Sivakumar <suharshs@google.com> | 2017-01-24 15:58:55 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-24 16:09:44 -0800 |
commit | d9ae30a17a0fea8666db00e13bcfb760be998988 (patch) | |
tree | d4f9830288a45b611bd06394f89bdd353ff8f8ca /tensorflow/contrib/rnn/ops | |
parent | 5507516cc1aafff5a51472eb9e4571236a3a94fe (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.cc | 4 | ||||
-rw-r--r-- | tensorflow/contrib/rnn/ops/lstm_ops_test.cc | 6 |
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); } }; |