Skip to content

Commit

Permalink
Update tensorflow-2.0.0.patch
Browse files Browse the repository at this point in the history
  • Loading branch information
jkjung-avt committed Oct 4, 2019
1 parent ec5e636 commit 184612f
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
4 changes: 2 additions & 2 deletions install_tensorflow-2.0.0.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
set -e

script_path=$(realpath $0)
#patch_path=$(dirname $script_path)/tensorflow/tensorflow-2.0.0.patch
patch_path=$(dirname $script_path)/tensorflow/tensorflow-2.0.0.patch

chip_id=$(cat /sys/module/tegra_fuse/parameters/tegra_chip_id)
case ${chip_id} in
Expand Down Expand Up @@ -50,7 +50,7 @@ fi
tar xzvf tensorflow-2.0.0.tar.gz
cd tensorflow-2.0.0

#patch -N -p1 < $patch_path && echo "tensorflow-2.0.0 source tree appears to be patched already. Continue..."
patch -N -p1 < $patch_path && echo "tensorflow-2.0.0 source tree appears to be patched already. Continue..."

echo "** Configure and build tensorflow-2.0.0"
export TMP=/tmp
Expand Down
6 changes: 3 additions & 3 deletions tensorflow/tensorflow-2.0.0.patch
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
diff -Nrup a/tensorflow-2.0.0/third_party/nccl/build_defs.bzl.tpl b/tensorflow-2.0.0/third_party/nccl/build_defs.bzl.tpl
--- a/tensorflow-2.0.0/third_party/nccl/build_defs.bzl.tpl 2019-09-28 05:56:33.000000000 +0800
+++ b/tensorflow-2.0.0/third_party/nccl/build_defs.bzl.tpl 2019-10-04 12:51:56.663973578 +0800
diff -Nrup a/third_party/nccl/build_defs.bzl.tpl b/third_party/nccl/build_defs.bzl.tpl
--- a/third_party/nccl/build_defs.bzl.tpl 2019-09-28 05:56:33.000000000 +0800
+++ b/third_party/nccl/build_defs.bzl.tpl 2019-10-04 12:51:56.663973578 +0800
@@ -40,7 +40,7 @@ def _rdc_copts():
# The global functions can not have a lower register count than the
# device functions. This is enforced by setting a fixed register count.
Expand Down

0 comments on commit 184612f

Please sign in to comment.