diff options
author | Joel Hestness <jthestness@gmail.com> | 2017-12-26 18:51:40 -0800 |
---|---|---|
committer | drpngx <drpngx@users.noreply.github.com> | 2017-12-26 18:51:40 -0800 |
commit | f5a27328adafacb8d88bb62df835fc34cd7ed46c (patch) | |
tree | afb0b38b14b26351bd76be0d2d12b814c3c9159a /tensorflow/contrib/mpi_collectives/mpi_message.proto | |
parent | 1f94604944ce48036d8a5fd6f8b1b24ca36be953 (diff) |
mpi_collectives: Refactor to fix build issues (#15534)
* mpi_collectives: Refactor to fix build issues
After TF commit 5c7f9e3, the mpi_collectives package would no longer build
ops and kernels. This build issue caused mpi_collectives import to fail in
Python with the following error: "NameError: Could not find operator MPISize
in dynamic library mpi_collectives.so".
To fix this issue, add build targets to ensure both ops and kernels are built
Note, also refactored the build targets and directory structure to more
closely match other contrib packages.
* mpi_collectives: Minor BUILD fix
* Minor: Buildifier fix
* mpi_collectives: Correct preprocessor defines
* mpi_collectives: Clearer defines inclusion
Diffstat (limited to 'tensorflow/contrib/mpi_collectives/mpi_message.proto')
-rw-r--r-- | tensorflow/contrib/mpi_collectives/mpi_message.proto | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/mpi_collectives/mpi_message.proto b/tensorflow/contrib/mpi_collectives/mpi_message.proto index 7fa5e20301..afbce981ae 100644 --- a/tensorflow/contrib/mpi_collectives/mpi_message.proto +++ b/tensorflow/contrib/mpi_collectives/mpi_message.proto @@ -15,7 +15,7 @@ limitations under the License. syntax = "proto3"; -package tensorflow.contrib.mpi; +package tensorflow.contrib.mpi_collectives; import "tensorflow/core/framework/tensor_shape.proto"; import "tensorflow/core/framework/types.proto"; |