1 # Copyright 2017 The TensorFlow Authors. All Rights Reserved. 2 # 3 # Licensed under the Apache License, Version 2.0 (the "License"); 4 # you may not use this file except in compliance with the License. 5 # You may obtain a copy of the License at 6 # 7 # http://www.apache.org/licenses/LICENSE-2.0 8 # 9 # Unless required by applicable law or agreed to in writing, software 10 # distributed under the License is distributed on an "AS IS" BASIS, 11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 # See the License for the specific language governing permissions and 13 # limitations under the License. 14 # ============================================================================= 15 """Cloud TPU profiler package.""" 16 17 from __future__ import absolute_import 18 from __future__ import division 19 from __future__ import print_function 20 21 from setuptools import setup 22 23 _VERSION = '1.12.0' 24 25 CONSOLE_SCRIPTS = [ 26 'capture_tpu_profile=cloud_tpu_profiler.main:run_main', 27 ] 28 29 setup( 30 name='cloud_tpu_profiler', 31 version=_VERSION.replace('-', ''), 32 description='Trace and profile Cloud TPU performance', 33 long_description='Tools for capture TPU profile', 34 url='https://www.tensorflow.org/tfrc/', 35 author='Google Inc.', 36 author_email='packages (at] tensorflow.org', 37 packages=['cloud_tpu_profiler'], 38 package_data={ 39 'cloud_tpu_profiler': ['data/*'], 40 }, 41 entry_points={ 42 'console_scripts': CONSOLE_SCRIPTS, 43 }, 44 classifiers=[ 45 # How mature is this project? Common values are 46 # 3 - Alpha 47 # 4 - Beta 48 # 5 - Production/Stable 49 'Development Status :: 5 - Production/Stable', 50 'Intended Audience :: Developers', 51 'Intended Audience :: Education', 52 'Intended Audience :: Science/Research', 53 'License :: OSI Approved :: Apache Software License', 54 'Programming Language :: Python :: 2', 55 'Programming Language :: Python :: 2.7', 56 'Programming Language :: Python :: 3', 57 'Programming Language :: Python :: 3.4', 58 'Programming Language :: Python :: 3.5', 59 'Programming Language :: Python :: 3.6', 60 'Topic :: Scientific/Engineering', 61 'Topic :: Scientific/Engineering :: Mathematics', 62 'Topic :: Scientific/Engineering :: Artificial Intelligence', 63 'Topic :: Software Development', 64 'Topic :: Software Development :: Libraries', 65 'Topic :: Software Development :: Libraries :: Python Modules', 66 ], 67 license='Apache 2.0', 68 keywords='tensorflow performance tpu', 69 ) 70