Add patch for CUDA 11.3

Author: FloopCZ

tensorflow_cc/cmake/TensorflowBase.cmake

@@ -19,4 +19,7 @@ ExternalProject_Add(
           COMMAND mkdir -p "${CMAKE_CURRENT_BINARY_DIR}/tensorflow/bazel-bin/bin/tensorflow/include/src"
           COMMAND touch "${CMAKE_CURRENT_BINARY_DIR}/tensorflow/bazel-bin/tensorflow/include/src/__placeholder__.h"
           COMMAND touch "${CMAKE_CURRENT_BINARY_DIR}/tensorflow/bazel-bin/bin/tensorflow/include/src/__placeholder__.h"
+  # Fix compilation with CUDA 11.3.
+  PATCH_COMMAND cp "${CMAKE_CURRENT_SOURCE_DIR}/patches/v2.5.0-Fix-NCCL-build-failure-with-CUDA-11.3.patch" .
+        COMMAND patch -p1 < v2.5.0-Fix-NCCL-build-failure-with-CUDA-11.3.patch
 )

tensorflow_cc/patches/v2.5.0-Fix-NCCL-build-failure-with-CUDA-11.3.patch

@@ -0,0 +1,13 @@
+diff --git a/third_party/nccl/build_defs.bzl.tpl b/third_party/nccl/build_defs.bzl.tpl
+index ceb65e3b632f3..c875717403857 100644
+--- a/third_party/nccl/build_defs.bzl.tpl
++++ b/third_party/nccl/build_defs.bzl.tpl
+@@ -204,7 +204,7 @@ def _prune_relocatable_code_impl(ctx):
+             arguments = arguments,
+             mnemonic = "nvprune",
+         )
+-        output.append(outputs)
++        outputs.append(output)
+ 
+     return DefaultInfo(files = depset(outputs))
+