experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #29673 from bas-aarts:cuda-mpi-compiler-error

Browse Source 
PiperOrigin-RevId: 253307779
This commit is contained in:
TensorFlower Gardener 2019-06-14 15:33:13 -07:00
commit 56968ba2f5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/contrib/mpi_collectives/kernels/ring.cu.cc
View File

@ -20,6 +20,7 @@ limitations under the License.
#define EIGEN_USE_GPU
#include "tensorflow/contrib/mpi_collectives/kernels/ring.h"
#include "tensorflow/core/util/gpu_kernel_helper.h"
#include "tensorflow/core/util/gpu_launch_config.h"
namespace tensorflow {