chore: import upstream snapshot with attribution

tensorflow/python/tpu/profiler/pip_package/BUILD

@@ -0,0 +1,15 @@
+# Description:
+#  Tools for building the Cloud TPU profiler pip package.
+
+package(default_visibility = ["//visibility:private"])
+
+licenses(["notice"])  # Apache 2.0
+
+sh_binary(
+    name = "build_pip_package",
+    srcs = ["build_pip_package.sh"],
+    data = [
+        "setup.py",
+        "//tensorflow/python/tpu/profiler:capture_tpu_profile_bin",
+    ],
+)

tensorflow/python/tpu/profiler/pip_package/README

@@ -0,0 +1,2 @@
+This package contains a Python wrapper around Tensorflow Python APIs that are
+used to profile Cloud TPU.

tensorflow/python/tpu/profiler/pip_package/build_pip_package.sh

@@ -0,0 +1,65 @@
+#!/usr/bin/env bash
+# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# =============================================================================
+
+
+set -e
+
+if [ "$(uname)" = "Darwin" ]; then
+  sedi="sed -i ''"
+else
+  sedi="sed -i"
+fi
+
+PACKAGE_NAME="cloud_tpu_profiler"
+PIP_PACKAGE="tensorflow/python/tpu/profiler/pip_package"
+RUNFILES="bazel-bin/tensorflow/python/tpu/profiler/pip_package/build_pip_package.runfiles/org_tensorflow/tensorflow/python/tpu/profiler"
+
+function main() {
+  if [ $# -lt 1 ] ; then
+    echo "No destination dir provided"
+    exit 1
+  fi
+
+  DEST=$1
+  TMPDIR=$(mktemp -d -t tmp.XXXXXXXXXX)
+
+  echo $(date) : "=== Using tmpdir: ${TMPDIR}"
+
+  cp ${PIP_PACKAGE}/README ${TMPDIR}
+  cp ${PIP_PACKAGE}/setup.py ${TMPDIR}
+  mkdir ${TMPDIR}/${PACKAGE_NAME}
+  cp -a ${RUNFILES}/. ${TMPDIR}/${PACKAGE_NAME}/
+
+  # Fix the import statements to reflect the copied over path.
+  find  ${TMPDIR}/${PACKAGE_NAME} -name \*.py |
+    xargs $sedi -e '
+      s/^from tensorflow.python.tpu.profiler/from '${PACKAGE_NAME}'/
+  '
+  echo $(ls $TMPDIR)
+
+  pushd ${TMPDIR}
+  echo $(date) : "=== Building wheel"
+  echo $(pwd)
+  python setup.py bdist_wheel >/dev/null
+  python3 setup.py bdist_wheel >/dev/null
+  mkdir -p ${DEST}
+  cp dist/* ${DEST}
+  popd
+  rm -rf ${TMPDIR}
+  echo $(date) : "=== Output wheel file is in: ${DEST}"
+}
+
+main "$@"

tensorflow/python/tpu/profiler/pip_package/setup.py

@@ -0,0 +1,58 @@
+# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# =============================================================================
+"""Cloud TPU profiler package."""
+
+from setuptools import setup
+
+from cloud_tpu_profiler.version import __version__
+
+CONSOLE_SCRIPTS = [
+    'capture_tpu_profile=cloud_tpu_profiler.capture_tpu_profile:run_main',
+]
+
+setup(
+    name='cloud_tpu_profiler',
+    version=__version__.replace('-', ''),
+    description='Trace and profile Cloud TPU performance',
+    long_description='Tools for capture TPU profile',
+    url='https://www.tensorflow.org/tfrc/',
+    author='Google Inc.',
+    author_email='packages@tensorflow.org',
+    packages=['cloud_tpu_profiler'],
+    entry_points={
+        'console_scripts': CONSOLE_SCRIPTS,
+    },
+    classifiers=[
+        'Development Status :: 5 - Production/Stable',
+        'Intended Audience :: Developers',
+        'Intended Audience :: Education',
+        'Intended Audience :: Science/Research',
+        'License :: OSI Approved :: Apache Software License',
+        'Programming Language :: Python :: 2',
+        'Programming Language :: Python :: 2.7',
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.4',
+        'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
+        'Topic :: Scientific/Engineering',
+        'Topic :: Scientific/Engineering :: Mathematics',
+        'Topic :: Scientific/Engineering :: Artificial Intelligence',
+        'Topic :: Software Development',
+        'Topic :: Software Development :: Libraries',
+        'Topic :: Software Development :: Libraries :: Python Modules',
+    ],
+    license='Apache 2.0',
+    keywords='tensorflow performance tpu',
+)