File fix-mpi-build.patch of Package tensorflow

Overview Repositories Revisions Requests Users Attributes Meta

File fix-mpi-build.patch of Package tensorflow

From ab02e2a5b343db8a128907a04e374c49141d5e3c Mon Sep 17 00:00:00 2001
From: Christian Goll <cgoll@suse.de>
Date: Mon, 16 Mar 2020 15:49:59 +0100
Subject: [PATCH] fix mpi build
 https://github.com/tensorflow/tensorflow/issues/17437

---
 tensorflow/contrib/mpi_collectives/BUILD | 1 +
 tensorflow/tensorflow.bzl                | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/tensorflow/contrib/mpi_collectives/BUILD b/tensorflow/contrib/mpi_collectives/BUILD
index f8072ac1e8..8f2649eb50 100644
--- a/tensorflow/contrib/mpi_collectives/BUILD
+++ b/tensorflow/contrib/mpi_collectives/BUILD
@@ -53,6 +53,7 @@ tf_custom_op_library(
         ":mpi_defines",
         ":mpi_message_proto_cc",
         "//third_party/mpi",
+        "//tensorflow/stream_executor",
     ],
 )
 
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 4aa412180f..d3e0a48c0d 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -1806,7 +1806,7 @@ def tf_custom_op_library(name, srcs = [], gpu_srcs = [], deps = [], linkopts = [
         name = name + "_check_deps",
         disallowed_deps = [
             clean_dep("//tensorflow/core:framework"),
-            clean_dep("//tensorflow/core:lib"),
+            #clean_dep("//tensorflow/core:lib"),
         ],
         deps = deps + if_cuda_is_configured_compat(cuda_deps) + if_rocm_is_configured(rocm_deps),
     )
-- 
2.25.0

Places

File fix-mpi-build.patch of Package tensorflow

Places