Skip to content

Commit

Permalink
Update XRTServer Tensorflow version
Browse files Browse the repository at this point in the history
  • Loading branch information
staticfloat committed Oct 24, 2018
1 parent 2081fde commit aa31f0a
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 3 deletions.
11 changes: 8 additions & 3 deletions XRTServer/build_tarballs.jl
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,12 @@ ENV["BINARYBUILDER_USE_CCACHE"] = "false"
using BinaryBuilder

name = "XRTServer"
version = v"2018.10.16"
version = v"2018.10.23"

# Collection of sources required
sources = [
"https://github.com/JuliaComputing/tensorflow.git" =>
"11b6ca9bc3506443c5b46f42f193ee7ba6fb1f50",
"36c34bd2d744f0027c24e3afae118b5d956ce741",
"https://github.com/bazelbuild/bazel/releases/download/0.17.2/bazel-0.17.2-linux-x86_64" =>
"674757d40d4ac0f0175df7fe84cd7250cbf67ac7ebac565e905fdc7e24c0fac5",

Expand Down Expand Up @@ -59,6 +59,9 @@ mv build/lib/* $prefix/lib/
cd $WORKSPACE/srcdir/tensorflow
atomic_patch -p0 $WORKSPACE/srcdir/patches/link_against_librt.patch
# Apply XRT patch
atomic_patch -p1 $WORKSPACE/srcdir/patches/xrt.patch
# Get `bazel` onto our $PATH
chmod +x $WORKSPACE/srcdir/bazel-*
mv $WORKSPACE/srcdir/bazel-* $WORKSPACE/srcdir/bazel
Expand All @@ -84,7 +87,9 @@ cp bazel-bin/tensorflow/libtensorflow_framework.so $prefix/lib/
cp bazel-bin/tensorflow/compiler/xrt/utils/xrt_server $prefix/bin/
# Cleanup things we don't need
rm -rf ${prefix}/{doc,jre,samples,nsight,libnvvp}
rm -rf ${prefix}/{doc,jre,samples,nsight,nsightee_plugins,libnvvp,libnsight}
rm -f ${prefix}/lib64/*.a
rm -f ${prefix}/lib/*.a
"""

# We attempt to build for only x86_64-linux-gnu
Expand Down
13 changes: 13 additions & 0 deletions XRTServer/bundled/patches/xrt.patch
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
diff --git a/tensorflow/BUILD b/tensorflow/BUILD
index 9b62a50452..63eed1fb5e 100644
--- a/tensorflow/BUILD
+++ b/tensorflow/BUILD
@@ -442,6 +442,8 @@ tf_cc_shared_object(
"//tensorflow/c:exported_symbols.lds",
"//tensorflow/c:version_script.lds",
"//tensorflow/c/eager:c_api",
+ "//tensorflow/compiler/xrt:xrt_server",
+ "//tensorflow/core/distributed_runtime/rpc:grpc_session",
"//tensorflow/core:tensorflow",
],
)

0 comments on commit aa31f0a

Please sign in to comment.