setup.py.in 3.8 KB
Newer Older
T
typhoonzero 已提交
1
from setuptools import setup, Distribution, Extension
Y
Yancey 已提交
2
import subprocess
3 4 5
class BinaryDistribution(Distribution):
    def has_ext_modules(foo):
        return True
Z
zhangjinchao01 已提交
6

Y
Yancey 已提交
7
MAJOR   = 0
T
typhoonzero 已提交
8
MINOR   = 11
Y
Yancey 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
PATCH   = 0
RC      = 0
ISTAGED = False



def git_commit():
    try:
        cmd = ['git', 'rev-parse', 'HEAD']
        git_commit = subprocess.Popen(cmd, stdout = subprocess.PIPE).communicate()[0].strip()
    except:
        git_commit = 'Unknown'
    return git_commit

def write_version_py(filename='paddle/version.py'):
    cnt = '''
# THIS FILE IS GENERATED FROM PADDLEPADDLE SETUP.PY
#
full_version    = '%(major)d.%(minor)d.%(patch)d'
major           = '%(major)d'
minor           = '%(minor)d'
patch           = '%(patch)d'
rc              = '%(rc)d'
istaged         = %(istaged)s
commit          = '%(commit)s'
L
Luo Tao 已提交
34
with_mkl        = '%(with_mkl)s'
Y
Yancey 已提交
35 36 37 38 39 40 41 42 43 44

def show():
    if istaged:
        print 'full_version:', full_version
        print 'major:', major
        print 'minor:', minor
        print 'patch:', patch
        print 'rc:', rc
    else:
        print 'commit:', commit
L
Luo Tao 已提交
45 46 47

def mkl():
    return with_mkl
Y
Yancey 已提交
48 49 50 51 52 53 54 55 56 57
'''
    commit = git_commit()
    with open(filename, 'w') as f:
        f.write(cnt % {
            'major': MAJOR,
            'minor': MINOR,
            'patch': PATCH,
            'rc': RC,
            'version': '${PADDLE_VERSION}',
            'commit': commit,
L
Luo Tao 已提交
58 59
            'istaged': ISTAGED,
            'with_mkl': '@WITH_MKL@'})
Y
Yancey 已提交
60

61
write_version_py(filename='@PADDLE_BINARY_DIR@/python/paddle/version.py')
Y
Yancey 已提交
62 63


Z
zhangjinchao01 已提交
64
packages=['paddle',
Q
qiaolongfei 已提交
65
          'paddle.utils',
66 67
          'paddle.dataset',
          'paddle.reader',
68 69
          'paddle.fluid',
          'paddle.fluid.proto',
X
Xin Pan 已提交
70
          'paddle.fluid.proto.profiler',
L
Luo Tao 已提交
71 72
          'paddle.fluid.layers']

73
if '${WITH_FLUID_ONLY}'== 'OFF':
L
Luo Tao 已提交
74 75 76 77 78 79
    packages+=['paddle.proto',
               'paddle.trainer',
               'paddle.trainer_config_helpers',
               'paddle.v2',
               'paddle.v2.master',
               'paddle.v2.plot',
80 81
               'paddle.v2.reader',
               'paddle.v2.dataset',
L
Luo Tao 已提交
82
               'py_paddle']
Z
zhangjinchao01 已提交
83

84 85
with open('@PADDLE_SOURCE_DIR@/python/requirements.txt') as f:
    setup_requires = f.read().splitlines()
86 87

if '${CMAKE_SYSTEM_PROCESSOR}' not in ['arm', 'armv7-a', 'aarch64']:
Y
Yancey 已提交
88
    setup_requires+=['opencv-python']
89

90
# the prefix is sys.prefix which should always be usr
L
Luo Tao 已提交
91
paddle_bins = ''
92
if '${WITH_FLUID_ONLY}'== 'OFF':
L
Luo Tao 已提交
93 94 95 96 97 98 99
    paddle_bin_dir = 'opt/paddle/bin'
    paddle_bins = ['${PADDLE_BINARY_DIR}/paddle/trainer/paddle_trainer',
                   '${PADDLE_BINARY_DIR}/paddle/trainer/paddle_merge_model',
                   '${PADDLE_BINARY_DIR}/paddle/pserver/paddle_pserver_main',
                   '${PADDLE_BINARY_DIR}/paddle/scripts/paddle']

package_data={'paddle.fluid': ['core.so']}
100
if '${WITH_FLUID_ONLY}'== 'OFF':
L
Luo Tao 已提交
101 102 103 104 105 106 107 108 109 110
    package_data['paddle.v2.master']=['libpaddle_master.so']
    package_data['py_paddle']=['*.py','_swig_paddle.so']

package_dir={
    '': '${CMAKE_CURRENT_SOURCE_DIR}',
    # The paddle.fluid.proto will be generated while compiling.
    # So that package points to other directory.
    'paddle.fluid.proto.profiler': '${PADDLE_BINARY_DIR}/paddle/fluid/platform',
    'paddle.fluid.proto': '${PADDLE_BINARY_DIR}/paddle/fluid/framework',
}
111
if '${WITH_FLUID_ONLY}'== 'OFF':
112
    package_dir['py_paddle']='${PADDLE_BINARY_DIR}/python/py_paddle'
L
Luo Tao 已提交
113
    
114

115
paddle_rt_lib_dir = 'lib'
L
Luo Tao 已提交
116 117 118
paddle_rt_libs = ['${WARPCTC_LIBRARIES}']
if '${MKL_SHARED_LIBS}'!= '':
  paddle_rt_libs += '${MKL_SHARED_LIBS}'.split(';')
T
tensor-tang 已提交
119

T
typhoonzero 已提交
120
setup(name='${PACKAGE_NAME}',
Z
zhangjinchao01 已提交
121 122
      version='${PADDLE_VERSION}',
      description='Parallel Distributed Deep Learning',
123
      install_requires=setup_requires,
L
Luo Tao 已提交
124
      packages=packages,
T
typhoonzero 已提交
125
      ext_modules=[Extension('_foo', ['stub.cc'])],
L
Luo Tao 已提交
126 127
      package_data=package_data,
      package_dir=package_dir,
128 129
      scripts=paddle_bins,
      data_files=[(paddle_rt_lib_dir, paddle_rt_libs)]
Z
zhangjinchao01 已提交
130
)