• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1diff --git a/tensorflow/tools/pip_package/build_pip_package.sh b/tensorflow/tools/pip_package/build_pip_package.sh
2index 305033cd44..d2c43dc7ab 100755
3--- a/tensorflow/tools/pip_package/build_pip_package.sh
4+++ b/tensorflow/tools/pip_package/build_pip_package.sh
5@@ -135,6 +135,7 @@ function prepare_src() {
6   cp tensorflow/tools/pip_package/MANIFEST.in ${TMPDIR}
7   cp tensorflow/tools/pip_package/README ${TMPDIR}
8   cp tensorflow/tools/pip_package/setup.py ${TMPDIR}
9+  touch ${TMPDIR}/stub.cc
10
11   rm -f ${TMPDIR}/tensorflow/libtensorflow_framework.so
12   rm -f ${TMPDIR}/tensorflow/libtensorflow_framework.so.[0-9].*
13diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
14index c36a2a0fa8..6ed096aafc 100644
15--- a/tensorflow/tools/pip_package/setup.py
16+++ b/tensorflow/tools/pip_package/setup.py
17@@ -34,7 +34,7 @@ import os
18 import re
19 import sys
20
21-from setuptools import Command
22+from setuptools import Command, Extension
23 from setuptools import find_packages
24 from setuptools import setup
25 from setuptools.command.install import install as InstallCommandBase
26@@ -269,6 +269,7 @@ setup(
27             EXTENSION_NAME,
28         ] + matches,
29     },
30+    ext_modules=[Extension('_foo', ['stub.cc'])],
31     zip_safe=False,
32     distclass=BinaryDistribution,
33     cmdclass={
34)
35