setup.py.in 27.5 KB
Newer Older
Y
Yancey 已提交
1
import subprocess
2
import os
3
import os.path
4
import errno
5
import re
6
import shutil
7
import sys
8
import fnmatch
C
Chengmo 已提交
9
import errno
10
import platform
11

12
from contextlib import contextmanager
13 14 15 16 17
from setuptools import Command
from setuptools import setup, Distribution, Extension
from setuptools.command.install import install as InstallCommandBase


18 19 20
class BinaryDistribution(Distribution):
    def has_ext_modules(foo):
        return True
Z
zhangjinchao01 已提交
21

Y
Yancey 已提交
22 23
RC      = 0

24
ext_name = '.dll' if os.name == 'nt' else ('.dylib' if sys.platform == 'darwin' else '.so')
Y
Yancey 已提交
25 26 27 28

def git_commit():
    try:
        cmd = ['git', 'rev-parse', 'HEAD']
29 30
        git_commit = subprocess.Popen(cmd, stdout = subprocess.PIPE,
            cwd="@PADDLE_SOURCE_DIR@").communicate()[0].strip()
Y
Yancey 已提交
31 32
    except:
        git_commit = 'Unknown'
33
    git_commit = git_commit.decode()
34
    return str(git_commit)
Y
Yancey 已提交
35

36
def _get_version_detail(idx):
37 38
    assert idx < 3, "vesion info consists of %(major)d.%(minor)d.%(patch)d, \
        so detail index must less than 3"
39

M
minqiyang 已提交
40 41
    if re.match('@TAG_VERSION_REGEX@', '@PADDLE_VERSION@'):
        version_details = '@PADDLE_VERSION@'.split('.')
42

M
minqiyang 已提交
43
        if len(version_details) >= 3:
M
minqiyang 已提交
44
            return version_details[idx]
45

46
    return 0
47

M
minqiyang 已提交
48
def get_major():
49
    return int(_get_version_detail(0))
50

M
minqiyang 已提交
51
def get_minor():
52
    return int(_get_version_detail(1))
53

54
def get_patch():
55
    return str(_get_version_detail(2))
56

57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
def get_cuda_version():
    if '@WITH_GPU@' == 'ON':
        return '@CUDA_VERSION@'
    else:
        return 'False'

def get_cudnn_version():
    if '@WITH_GPU@' == 'ON':
        temp_cudnn_version = ''
        if '@CUDNN_MAJOR_VERSION@':
            temp_cudnn_version += '@CUDNN_MAJOR_VERSION@'
            if '@CUDNN_MINOR_VERSION@':
                temp_cudnn_version += '.@CUDNN_MINOR_VERSION@'
                if '@CUDNN_PATCHLEVEL_VERSION@':
                    temp_cudnn_version += '.@CUDNN_PATCHLEVEL_VERSION@'
        return temp_cudnn_version
    else:
        return 'False'

76
def is_taged():
M
minqiyang 已提交
77
    try:
78
        cmd = ['git', 'describe', '--exact-match', '--tags', 'HEAD', '2>/dev/null']
79
        git_tag = subprocess.Popen(cmd, stdout = subprocess.PIPE, cwd="@PADDLE_SOURCE_DIR@").communicate()[0].strip()
80
        git_tag = git_tag.decode()
M
minqiyang 已提交
81
    except:
82 83
        return False

84
    if str(git_tag).replace('v', '') == '@PADDLE_VERSION@':
85 86
        return True
    else:
M
minqiyang 已提交
87
        return False
88

89
def write_version_py(filename='paddle/version/__init__.py'):
90
    cnt = '''# THIS FILE IS GENERATED FROM PADDLEPADDLE SETUP.PY
Y
Yancey 已提交
91
#
92
full_version    = '%(major)d.%(minor)d.%(patch)s'
Y
Yancey 已提交
93 94
major           = '%(major)d'
minor           = '%(minor)d'
95
patch           = '%(patch)s'
Y
Yancey 已提交
96
rc              = '%(rc)d'
97 98
cuda_version    = '%(cuda)s'
cudnn_version   = '%(cudnn)s'
Y
Yancey 已提交
99 100
istaged         = %(istaged)s
commit          = '%(commit)s'
L
Luo Tao 已提交
101
with_mkl        = '%(with_mkl)s'
Y
Yancey 已提交
102

103
__all__ = ['cuda', 'cudnn', 'show']
104

Y
Yancey 已提交
105
def show():
106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
    """Get the version of paddle if `paddle` package if tagged. Otherwise, output the corresponding commit id.
    
    Returns:
        If paddle package is not tagged, the commit-id of paddle will be output.
        Otherwise, the following information will be output.

        full_version: version of paddle

        major: the major version of paddle

        minor: the minor version of paddle

        patch: the patch level version of paddle
        
        rc: whether it's rc version

        cuda: the cuda version of package. It will return `False` if CPU version paddle package is installed

        cudnn: the cudnn version of package. It will return `False` if CPU version paddle package is installed
    
    Examples:
        .. code-block:: python

            import paddle

            # Case 1: paddle is tagged with 2.2.0
            paddle.version.show()
            # full_version: 2.2.0
            # major: 2
            # minor: 2
            # patch: 0
            # rc: 0
            # cuda: '10.2'
            # cudnn: '7.6.5'

            # Case 2: paddle is not tagged
            paddle.version.show()
            # commit: cfa357e984bfd2ffa16820e354020529df434f7d
            # cuda: '10.2'
            # cudnn: '7.6.5'
    """
Y
Yancey 已提交
147
    if istaged:
148 149 150 151 152
        print('full_version:', full_version)
        print('major:', major)
        print('minor:', minor)
        print('patch:', patch)
        print('rc:', rc)
Y
Yancey 已提交
153
    else:
154
        print('commit:', commit)
155 156
    print('cuda:', cuda_version)
    print('cudnn:', cudnn_version)
L
Luo Tao 已提交
157 158 159

def mkl():
    return with_mkl
160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193

def cuda():
    """Get cuda version of paddle package.

    Returns:
        string: Return the version information of cuda. If paddle package is CPU version, it will return False.
    
    Examples:
        .. code-block:: python

            import paddle

            paddle.version.cuda()
            # '10.2'

    """
    return cuda_version

def cudnn():
    """Get cudnn version of paddle package.

    Returns:
        string: Return the version information of cudnn. If paddle package is CPU version, it will return False.
    
    Examples:
        .. code-block:: python

            import paddle

            paddle.version.cudnn()
            # '7.6.5'

    """
    return cudnn_version
Y
Yancey 已提交
194 195
'''
    commit = git_commit()
196 197 198 199 200 201 202 203 204

    dirname = os.path.dirname(filename)

    try:
        os.makedirs(dirname)
    except OSError as e:
        if e.errno != errno.EEXIST:
            raise

Y
Yancey 已提交
205 206
    with open(filename, 'w') as f:
        f.write(cnt % {
207 208 209
            'major': get_major(),
            'minor': get_minor(),
            'patch': get_patch(),
Y
Yancey 已提交
210 211
            'rc': RC,
            'version': '${PADDLE_VERSION}',
212 213
            'cuda': get_cuda_version(),
            'cudnn': get_cudnn_version(),
Y
Yancey 已提交
214
            'commit': commit,
215
            'istaged': is_taged(),
L
Luo Tao 已提交
216
            'with_mkl': '@WITH_MKL@'})
Y
Yancey 已提交
217

218
write_version_py(filename='@PADDLE_BINARY_DIR@/python/paddle/version/__init__.py')
Y
Yancey 已提交
219

220 221 222 223 224 225 226 227 228 229 230 231 232 233
def write_cuda_env_config_py(filename='paddle/cuda_env.py'):
    cnt = ""
    if '${JIT_RELEASE_WHL}' == 'ON':
        cnt = '''# THIS FILE IS GENERATED FROM PADDLEPADDLE SETUP.PY
#
import os
os.environ['CUDA_CACHE_MAXSIZE'] = '805306368'
'''

    with open(filename, 'w') as f:
        f.write(cnt)

write_cuda_env_config_py(filename='@PADDLE_BINARY_DIR@/python/paddle/cuda_env.py')

C
Chengmo 已提交
234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261
def write_distributed_training_mode_py(filename='paddle/fluid/incubate/fleet/parameter_server/version.py'):
    cnt = '''from __future__ import print_function

# THIS FILE IS GENERATED FROM PADDLEPADDLE SETUP.PY

from paddle.fluid.incubate.fleet.base.mode import Mode

BUILD_MODE=Mode.%(mode)s

def is_transpiler():
    return Mode.TRANSPILER == BUILD_MODE

'''

    dirname = os.path.dirname(filename)

    try:
        os.makedirs(dirname)
    except OSError as e:
        if e.errno != errno.EEXIST:
            raise

    with open(filename, 'w') as f:
        f.write(cnt % {
            'mode': 'PSLIB' if '${WITH_PSLIB}' == 'ON' else 'TRANSPILER'
        })

write_distributed_training_mode_py(filename='@PADDLE_BINARY_DIR@/python/paddle/fluid/incubate/fleet/parameter_server/version.py')
Y
Yancey 已提交
262

263

Z
zhangjinchao01 已提交
264
packages=['paddle',
265
          'paddle.libs',
Q
qiaolongfei 已提交
266
          'paddle.utils',
267
          'paddle.utils.gast',
268
          'paddle.utils.cpp_extension',
269 270
          'paddle.dataset',
          'paddle.reader',
271
          'paddle.distributed',
272
          'paddle.distributed.metric',
Z
ziyoujiyi 已提交
273 274
          'paddle.distributed.ps',
          'paddle.distributed.ps.utils',
275
          'paddle.incubate',
276
          'paddle.incubate.autograd',
277
          'paddle.incubate.optimizer',
278
          'paddle.incubate.checkpoint',
279
          'paddle.incubate.operators',
280
          'paddle.incubate.tensor',
Z
Zhong Hui 已提交
281
          'paddle.incubate.multiprocessing',
282
          'paddle.incubate.nn',
W
wuhuanzhou 已提交
283
          'paddle.incubate.passes',
284
          'paddle.distribution',
B
Baibaifan 已提交
285
          'paddle.distributed.sharding',
286
          'paddle.distributed.fleet',
287 288 289 290 291 292
          'paddle.distributed.launch',
          'paddle.distributed.launch.context',
          'paddle.distributed.launch.controllers',
          'paddle.distributed.launch.job',
          'paddle.distributed.launch.plugins',
          'paddle.distributed.launch.utils',
293
          'paddle.distributed.fleet.base',
K
kuizhiqing 已提交
294
          'paddle.distributed.fleet.elastic',
295
          'paddle.distributed.fleet.meta_optimizers',
296
          'paddle.distributed.fleet.meta_optimizers.sharding',
297
          'paddle.distributed.fleet.meta_optimizers.ascend',
298
          'paddle.distributed.fleet.meta_optimizers.dygraph_optimizer',
299 300
          'paddle.distributed.fleet.runtime',
          'paddle.distributed.fleet.dataset',
301
          'paddle.distributed.fleet.data_generator',
302 303 304
          'paddle.distributed.fleet.metrics',
          'paddle.distributed.fleet.proto',
          'paddle.distributed.fleet.utils',
305
          'paddle.distributed.fleet.meta_parallel',
306
          'paddle.distributed.fleet.meta_parallel.pp_utils',
B
Baibaifan 已提交
307
          'paddle.distributed.fleet.meta_parallel.sharding',
308
          'paddle.distributed.fleet.meta_parallel.parallel_layers',
309
          'paddle.distributed.auto_parallel',
310
          'paddle.distributed.auto_parallel.operators',
311
          'paddle.distributed.auto_parallel.tuner',
312
          'paddle.distributed.auto_parallel.cost',
313
          'paddle.distributed.passes',
314
          'paddle.framework',
315
          'paddle.jit',
316
          'paddle.jit.dy2static',
W
Wilber 已提交
317
          'paddle.inference',
318 319
          'paddle.inference.contrib',
          'paddle.inference.contrib.utils',
320
          'paddle.fluid',
W
Wilber 已提交
321
          'paddle.fluid.inference',
L
lujun 已提交
322
          'paddle.fluid.dygraph',
323
          'paddle.fluid.dygraph.dygraph_to_static',
324
          'paddle.fluid.dygraph.amp',
325
          'paddle.fluid.proto',
X
Xin Pan 已提交
326
          'paddle.fluid.proto.profiler',
H
heqiaozhi 已提交
327
          'paddle.fluid.distributed',
Y
Yancey 已提交
328
          'paddle.fluid.layers',
329
          'paddle.fluid.dataloader',
Q
Qingsheng Li 已提交
330 331
          'paddle.fluid.contrib',
          'paddle.fluid.contrib.decoder',
D
Dang Qingqing 已提交
332
          'paddle.fluid.contrib.quantize',
W
whs 已提交
333
          'paddle.fluid.contrib.slim',
W
WangZhen 已提交
334
          'paddle.fluid.contrib.slim.quantization',
335
          'paddle.fluid.contrib.slim.quantization.imperative',
C
chengduo 已提交
336
          'paddle.fluid.contrib.extend_optimizer',
337
          'paddle.fluid.contrib.mixed_precision',
338
          'paddle.fluid.contrib.mixed_precision.bf16',
339
          'paddle.fluid.contrib.layers',
340
          'paddle.fluid.contrib.sparsity',
Q
qiaolongfei 已提交
341
          'paddle.fluid.transpiler',
D
dongdaxiang 已提交
342 343
          'paddle.fluid.transpiler.details',
          'paddle.fluid.incubate',
344
          'paddle.fluid.incubate.data_generator',
D
dongdaxiang 已提交
345
          'paddle.fluid.incubate.fleet',
346
          'paddle.fluid.incubate.checkpoint',
D
dongdaxiang 已提交
347 348
          'paddle.fluid.incubate.fleet.base',
          'paddle.fluid.incubate.fleet.parameter_server',
T
tangwei12 已提交
349
          'paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler',
350
          'paddle.fluid.incubate.fleet.parameter_server.pslib',
351
          'paddle.fluid.incubate.fleet.parameter_server.ir',
352
          'paddle.fluid.incubate.fleet.collective',
353
          'paddle.fluid.incubate.fleet.utils',
354
          'paddle.amp',
H
huangxu96 已提交
355
          'paddle.cost_model',
356 357 358 359 360 361 362
          'paddle.hapi',
          'paddle.vision',
          'paddle.vision.models',
          'paddle.vision.transforms',
          'paddle.vision.datasets',
          'paddle.text',
          'paddle.text.datasets',
H
hong 已提交
363
          'paddle.incubate',
364 365 366
          'paddle.incubate.nn',
          'paddle.incubate.nn.functional',
          'paddle.incubate.nn.layer',
367
          'paddle.incubate.optimizer.functional',
368
          'paddle.io',
H
hong 已提交
369
          'paddle.optimizer',
370 371
          'paddle.nn',
          'paddle.nn.functional',
372
          'paddle.nn.layer',
373
          'paddle.nn.quant',
H
hong 已提交
374
          'paddle.nn.initializer',
375
          'paddle.nn.utils',
H
hong 已提交
376
          'paddle.metric',
377 378
          'paddle.static',
          'paddle.static.nn',
379
          'paddle.static.amp',
380
          'paddle.static.sparsity',
381
          'paddle.tensor',
C
channings 已提交
382
          'paddle.onnx',
383
          'paddle.autograd',
384 385
          'paddle.device',
          'paddle.device.cuda',
386
          'paddle.version',
C
chenjian 已提交
387
          'paddle.profiler'
388
          ]
L
Luo Tao 已提交
389

390 391
with open('@PADDLE_SOURCE_DIR@/python/requirements.txt') as f:
    setup_requires = f.read().splitlines()
392

393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416
# Note(wangzhongpu):
# When compiling paddle under python36, the dependencies belonging to python2.7 will be imported, resulting in errors when installing paddle
if sys.version_info >= (3,6) and sys.version_info < (3,7):
    setup_requires_tmp = []
    for setup_requires_i in setup_requires:
        if "<\"3.6\"" in setup_requires_i or "<\"3.5\"" in setup_requires_i or "<=\"3.5\"" in setup_requires_i:
            continue
        setup_requires_tmp+=[setup_requires_i]
    setup_requires = setup_requires_tmp
if sys.version_info >= (3,5) and sys.version_info < (3,6):
    setup_requires_tmp = []
    for setup_requires_i in setup_requires:
        if "<\"3.5\"" in setup_requires_i:
            continue
        setup_requires_tmp+=[setup_requires_i]
    setup_requires = setup_requires_tmp
if sys.version_info >= (3,7):
    setup_requires_tmp = []
    for setup_requires_i in setup_requires:
        if "<\"3.6\"" in setup_requires_i or "<=\"3.6\"" in setup_requires_i or "<\"3.5\"" in setup_requires_i or "<=\"3.5\"" in setup_requires_i or "<\"3.7\"" in setup_requires_i:
            continue
        setup_requires_tmp+=[setup_requires_i]
    setup_requires = setup_requires_tmp

417
# the prefix is sys.prefix which should always be usr
L
Luo Tao 已提交
418
paddle_bins = ''
419

T
Tao Luo 已提交
420 421
if not '${WIN32}':
    paddle_bins = ['${PADDLE_BINARY_DIR}/paddle/scripts/paddle']
422 423 424 425 426 427

if os.name != 'nt':
    package_data={'paddle.fluid': ['${FLUID_CORE_NAME}' + '.so']}
else:
    package_data={'paddle.fluid': ['${FLUID_CORE_NAME}' + '.pyd', '${FLUID_CORE_NAME}' + '.lib']}

H
huangxu96 已提交
428
package_data['paddle.fluid'] += ['${PADDLE_BINARY_DIR}/python/paddle/cost_model/static_op_benchmark.json']
429
if '${HAS_NOAVX_CORE}' == 'ON':
430
    package_data['paddle.fluid'] += ['core_noavx' + ('.so' if os.name != 'nt' else '.pyd')]
P
peizhilin 已提交
431

L
Luo Tao 已提交
432
package_dir={
433
    '': '${PADDLE_BINARY_DIR}/python',
L
Luo Tao 已提交
434 435 436 437
    # 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',
Q
qiaolongfei 已提交
438
    'paddle.fluid': '${PADDLE_BINARY_DIR}/python/paddle/fluid',
L
Luo Tao 已提交
439
}
440

441 442
# put all thirdparty libraries in paddle.libs
libs_path='${PADDLE_BINARY_DIR}/python/paddle/libs'
P
peizhilin 已提交
443

P
peizhilin 已提交
444
package_data['paddle.libs']= []
P
peizhilin 已提交
445
package_data['paddle.libs']=[('libwarpctc' if os.name != 'nt' else 'warpctc') + ext_name]
P
peizhilin 已提交
446
shutil.copy('${WARPCTC_LIBRARIES}', libs_path)
P
peizhilin 已提交
447

448 449 450 451 452 453 454 455 456 457 458 459 460
package_data['paddle.libs']+=[
    os.path.basename('${LAPACK_LIB}'), 
    os.path.basename('${BLAS_LIB}'),
    os.path.basename('${GFORTRAN_LIB}'),
    os.path.basename('${GNU_RT_LIB_1}')]
shutil.copy('${BLAS_LIB}', libs_path)
shutil.copy('${LAPACK_LIB}', libs_path)
shutil.copy('${GFORTRAN_LIB}', libs_path)
shutil.copy('${GNU_RT_LIB_1}', libs_path)
if not sys.platform.startswith("linux"):
    package_data['paddle.libs']+=[os.path.basename('${GNU_RT_LIB_2}')]
    shutil.copy('${GNU_RT_LIB_2}', libs_path)

461
if '${WITH_MKL}' == 'ON':
P
peizhilin 已提交
462 463 464
    shutil.copy('${MKLML_SHARED_LIB}', libs_path)
    shutil.copy('${MKLML_SHARED_IOMP_LIB}', libs_path)
    package_data['paddle.libs']+=[('libmklml_intel' if os.name != 'nt' else 'mklml') + ext_name, ('libiomp5' if os.name != 'nt' else 'libiomp5md') + ext_name]
P
peizhilin 已提交
465 466
else:
    if os.name == 'nt':
P
peizhilin 已提交
467
        # copy the openblas.dll
468
        shutil.copy('${OPENBLAS_SHARED_LIB}', libs_path)
P
peizhilin 已提交
469
        package_data['paddle.libs'] += ['openblas' + ext_name]
W
Wilber 已提交
470
    elif os.name == 'posix' and platform.machine() == 'aarch64' and '${OPENBLAS_LIB}'.endswith('so'):
H
houj04 已提交
471 472
        # copy the libopenblas.so on linux+aarch64
        # special: core_noavx.so depends on 'libopenblas.so.0', not 'libopenblas.so'
W
Wilber 已提交
473 474 475
        if os.path.exists('${OPENBLAS_LIB}' + '.0'):
            shutil.copy('${OPENBLAS_LIB}' + '.0', libs_path)
            package_data['paddle.libs'] += ['libopenblas.so.0']
P
peizhilin 已提交
476

477 478 479
if '${WITH_LITE}' == 'ON':
    shutil.copy('${LITE_SHARED_LIB}', libs_path)
    package_data['paddle.libs']+=['libpaddle_full_api_shared' + ext_name]
480 481 482 483 484 485
    if '${LITE_WITH_NNADAPTER}' == 'ON':
        shutil.copy('${LITE_NNADAPTER_LIB}', libs_path)
        package_data['paddle.libs']+=['libnnadapter' + ext_name]
        if '${NNADAPTER_WITH_HUAWEI_ASCEND_NPU}' == 'ON':
            shutil.copy('${LITE_NNADAPTER_NPU_LIB}', libs_path)
            package_data['paddle.libs']+=['libnnadapter_driver_huawei_ascend_npu' + ext_name]
486

487 488
if '${WITH_CINN}' == 'ON':
    shutil.copy('${CINN_LIB_LOCATION}/${CINN_LIB_NAME}', libs_path)
489
    shutil.copy('${CINN_INCLUDE_DIR}/cinn/runtime/cuda/cinn_cuda_runtime_source.cuh', libs_path)
490
    package_data['paddle.libs']+=['libcinnapi.so']
491
    package_data['paddle.libs']+=['cinn_cuda_runtime_source.cuh']
492

X
xujiaqi01 已提交
493 494
if '${WITH_PSLIB}' == 'ON':
    shutil.copy('${PSLIB_LIB}', libs_path)
X
xujiaqi01 已提交
495 496
    if os.path.exists('${PSLIB_VERSION_PY}'):
        shutil.copy('${PSLIB_VERSION_PY}', '${PADDLE_BINARY_DIR}/python/paddle/fluid/incubate/fleet/parameter_server/pslib/')
X
xujiaqi01 已提交
497 498
    package_data['paddle.libs'] += ['libps' + ext_name]

S
Sang Ik Lee 已提交
499
if '${WITH_MKLDNN}' == 'ON':
P
peizhilin 已提交
500
    if '${CMAKE_BUILD_TYPE}' == 'Release' and os.name != 'nt':
P
peizhilin 已提交
501 502 503 504
        # only change rpath in Release mode.
        # TODO(typhoonzero): use install_name_tool to patch mkl libs once
        # we can support mkl on mac.
        #
A
Adam 已提交
505
        # change rpath of libdnnl.so.1, add $ORIGIN/ to it.
P
peizhilin 已提交
506
        # The reason is that all thirdparty libraries in the same directory,
A
Adam 已提交
507
        # thus, libdnnl.so.1 will find libmklml_intel.so and libiomp5.so.
P
peizhilin 已提交
508 509
        command = "patchelf --set-rpath '$ORIGIN/' ${MKLDNN_SHARED_LIB}"
        if os.system(command) != 0:
A
Adam 已提交
510
            raise Exception("patch libdnnl.so failed, command: %s" % command)
S
Sang Ik Lee 已提交
511
    shutil.copy('${MKLDNN_SHARED_LIB}', libs_path)
A
Adam 已提交
512 513
    if os.name != 'nt':
        shutil.copy('${MKLDNN_SHARED_LIB_1}', libs_path)
514 515
        shutil.copy('${MKLDNN_SHARED_LIB_2}', libs_path)
        package_data['paddle.libs']+=['libmkldnn.so.0', 'libdnnl.so.1', 'libdnnl.so.2']
A
Adam 已提交
516 517 518
    else:
        package_data['paddle.libs']+=['mkldnn.dll']

519 520 521 522 523 524 525 526 527 528 529 530
if '${WITH_ONNXRUNTIME}' == 'ON':
    shutil.copy('${ONNXRUNTIME_SHARED_LIB}', libs_path)
    if os.name == 'nt':
        shutil.copy('${PADDLE2ONNX_SHARED_LIB}', libs_path)
        package_data['paddle.libs']+=['paddle2onnx.dll', 'onnxruntime.dll']
    else:
        shutil.copy('${PADDLE2ONNX_LIB}', libs_path)
        if sys.platform == 'darwin':
            package_data['paddle.libs']+=['libpaddle2onnx.dylib', 'libonnxruntime.1.10.0.dylib']
        else:
            package_data['paddle.libs']+=['libpaddle2onnx.so', 'libonnxruntime.so.1.10.0']

531 532 533 534 535 536 537 538 539 540 541 542 543
if '${WITH_XPU}' == 'ON':
    # only change rpath in Release mode,
    if '${CMAKE_BUILD_TYPE}' == 'Release':
        if os.name != 'nt':
            if "@APPLE@" == "1":
                command = "install_name_tool -id \"@loader_path/\" ${XPU_API_LIB}"
            else:
                command = "patchelf --set-rpath '$ORIGIN/' ${XPU_API_LIB}"
            if os.system(command) != 0:
                raise Exception("patch ${XPU_API_LIB} failed, command: %s" % command)
    shutil.copy('${XPU_API_LIB}', libs_path)
    shutil.copy('${XPU_RT_LIB}', libs_path)
    package_data['paddle.libs']+=['${XPU_API_LIB_NAME}',
544
                                  '${XPU_RT_LIB_NAME}']
545

546 547 548 549
if '${WITH_XPU_BKCL}' == 'ON':
    shutil.copy('${XPU_BKCL_LIB}', libs_path)
    package_data['paddle.libs']+=['${XPU_BKCL_LIB_NAME}']

550
# remove unused paddle/libs/__init__.py
P
peizhilin 已提交
551 552
if os.path.isfile(libs_path+'/__init__.py'):
    os.remove(libs_path+'/__init__.py')
553 554
package_dir['paddle.libs']=libs_path

555

556
# change rpath of ${FLUID_CORE_NAME}.ext, add $ORIGIN/../libs/ to it.
557
# The reason is that libwarpctc.ext, libiomp5.ext etc are in paddle.libs, and
558
# ${FLUID_CORE_NAME}.ext is in paddle.fluid, thus paddle/fluid/../libs will pointer to above libraries.
559
# This operation will fix https://github.com/PaddlePaddle/Paddle/issues/3213
L
luotao1 已提交
560
if '${CMAKE_BUILD_TYPE}' == 'Release':
P
peizhilin 已提交
561
    if os.name != 'nt':
562
        # only change rpath in Release mode, since in Debug mode, ${FLUID_CORE_NAME}.xx is too large to be changed.
L
luotao1 已提交
563
        if "@APPLE@" == "1":
564 565
            commands = ["install_name_tool -id '@loader_path/../libs/' ${PADDLE_BINARY_DIR}/python/paddle/fluid/${FLUID_CORE_NAME}" + '.so']
            commands.append("install_name_tool -add_rpath '@loader_path/../libs/' ${PADDLE_BINARY_DIR}/python/paddle/fluid/${FLUID_CORE_NAME}" + '.so')
L
luotao1 已提交
566
        else:
567
            commands = ["patchelf --set-rpath '$ORIGIN/../libs/' ${PADDLE_BINARY_DIR}/python/paddle/fluid/${FLUID_CORE_NAME}" + '.so']
W
Wilber 已提交
568
        # The sw_64 not suppot patchelf, so we just disable that.
H
houj04 已提交
569
        if platform.machine() != 'sw_64' and platform.machine() != 'mips64':
570 571 572
            for command in commands:
                if os.system(command) != 0:
                    raise Exception("patch ${FLUID_CORE_NAME}.%s failed, command: %s" % (ext_name, command))
P
peizhilin 已提交
573

P
peizhilin 已提交
574
ext_modules = [Extension('_foo', ['stub.cc'])]
P
peizhilin 已提交
575 576 577 578 579 580
if os.name == 'nt':
    # fix the path separator under windows
    fix_package_dir = {}
    for k, v in package_dir.items():
        fix_package_dir[k] = v.replace('/', '\\')
    package_dir = fix_package_dir
P
peizhilin 已提交
581
    ext_modules = []
582 583
elif sys.platform == 'darwin':
    ext_modules = []
T
tensor-tang 已提交
584

585
def find_files(pattern, root, recursive=False):
586
    for dirpath, _, files in os.walk(root):
587 588 589 590
        for filename in fnmatch.filter(files, pattern):
            yield os.path.join(dirpath, filename)
        if not recursive:
            break
591 592

headers = (
593
    # paddle level api headers
594
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle')) +
595
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/api')) +  # phi unify api header
596
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/api/ext')) +  # custom op api
597 598 599
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/api/include')) +  # phi api
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/common')) +  # phi common headers
    # phi level api headers (low level api)
600 601
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi')) +  # phi extension header
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/include', recursive=True)) +  # phi include headers
602
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends', recursive=True)) +  # phi backends headers
603 604 605
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/core', recursive=True)) +  # phi core headers
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/infermeta', recursive=True)) +  # phi infermeta headers
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/kernels', recursive=True)) +  # phi kernels headers
606
    # utila api headers
607
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/utils', recursive=True)))  # paddle utils headers
608

609 610 611
if '${WITH_MKLDNN}' == 'ON':
    headers += list(find_files('*', '${MKLDNN_INSTALL_DIR}/include')) # mkldnn

612
if '${WITH_GPU}' == 'ON' or '${WITH_ROCM}' == 'ON':
613 614
    # externalErrorMsg.pb for External Error message
    headers += list(find_files('*.pb', '${externalError_INCLUDE_DIR}'))
615 616 617 618 619

class InstallCommand(InstallCommandBase):
    def finalize_options(self):
        ret = InstallCommandBase.finalize_options(self)
        self.install_lib = self.install_platlib
620
        self.install_headers = os.path.join(self.install_platlib, 'paddle', 'include')
621
        return ret
622

623 624 625 626
class InstallHeaders(Command):
    """Override how headers are copied.
    """
    description = 'install C/C++ header files'
627

628 629 630 631 632
    user_options = [('install-dir=', 'd',
                     'directory to install header files to'),
                    ('force', 'f',
                     'force installation (overwrite existing files)'),
                   ]
633

634
    boolean_options = ['force']
635

636 637 638 639
    def initialize_options(self):
        self.install_dir = None
        self.force = 0
        self.outfiles = []
640

641 642 643 644
    def finalize_options(self):
        self.set_undefined_options('install',
                                   ('install_headers', 'install_dir'),
                                   ('force', 'force'))
645

646 647 648 649
    def mkdir_and_copy_file(self, header):
        if 'pb.h' in header:
            install_dir = re.sub('${PADDLE_BINARY_DIR}/', '', header)
        elif 'third_party' not in header:
650
            # paddle headers
651 652 653 654
            install_dir = re.sub('@PADDLE_SOURCE_DIR@/', '', header)
        else:
            # third_party
            install_dir = re.sub('${THIRD_PARTY_PATH}', 'third_party', header)
655
            patterns = ['install/mkldnn/include']
656 657
            for pattern in patterns:
                install_dir = re.sub(pattern, '', install_dir)
658 659 660 661
        install_dir = os.path.join(self.install_dir, os.path.dirname(install_dir))
        if not os.path.exists(install_dir):
            self.mkpath(install_dir)
        return self.copy_file(header, install_dir)
662

663 664 665 666 667 668 669 670
    def run(self):
        hdrs = self.distribution.headers
        if not hdrs:
            return
        self.mkpath(self.install_dir)
        for header in hdrs:
            (out, _) = self.mkdir_and_copy_file(header)
            self.outfiles.append(out)
671

672 673
    def get_inputs(self):
        return self.distribution.headers or []
674

675 676 677
    def get_outputs(self):
        return self.outfiles

678 679 680 681
# we redirect setuptools log for non-windows
if sys.platform != 'win32':
    @contextmanager
    def redirect_stdout():
682 683 684 685 686 687 688
        f_log = open('${SETUP_LOG_FILE}', 'w')
        origin_stdout = sys.stdout
        sys.stdout = f_log
        yield
        f_log = sys.stdout
        sys.stdout = origin_stdout
        f_log.close()
689 690 691 692 693
else:
    @contextmanager
    def redirect_stdout():
        yield

694 695
# Log for PYPI
if sys.version_info > (3,0):
696
    with open("@PADDLE_BINARY_DIR@/python/paddle/README.rst", "r", encoding='UTF-8') as f:
697 698
        long_description = f.read()
else:
699
    with open("@PADDLE_BINARY_DIR@/python/paddle/README.rst", "r")as f:
700 701
        long_description = unicode(f.read(), 'UTF-8')

W
wuhuanzhou 已提交
702 703 704 705 706 707
# strip *.so to reduce package size
if '${WITH_STRIP}' == 'ON':
    command = 'find ${PADDLE_BINARY_DIR}/python/paddle -name "*.so" | xargs -i strip {}'
    if os.system(command) != 0:
        raise Exception("strip *.so failed, command: %s" % command)

708 709 710 711
with redirect_stdout():
    setup(name='${PACKAGE_NAME}',
        version='${PADDLE_VERSION}',
        description='Parallel Distributed Deep Learning',
712 713 714 715 716 717 718 719 720 721
        long_description=long_description,
        long_description_content_type="text/markdown",
        author_email="Paddle-better@baidu.com",
        maintainer="PaddlePaddle",
        maintainer_email="Paddle-better@baidu.com",
        project_urls = {
            'Homepage': 'https://www.paddlepaddle.org.cn/',
            'Downloads': 'https://github.com/paddlepaddle/paddle'
        }, 
        license='Apache Software License',
722
        packages=packages,
723
        install_requires=setup_requires,
724 725 726 727 728 729 730 731 732
        ext_modules=ext_modules,
        package_data=package_data,
        package_dir=package_dir,
        scripts=paddle_bins,
        distclass=BinaryDistribution,
        headers=headers,
        cmdclass={
            'install_headers': InstallHeaders,
            'install': InstallCommand,
733 734 735
        },
        entry_points={
            'console_scripts': [
736
                'fleetrun = paddle.distributed.launch.main:launch'
737
            ]
738 739 740 741 742 743 744 745 746 747 748 749 750 751 752
        },
        classifiers=[
            'Development Status :: 5 - Production/Stable',
            'Operating System :: OS Independent',
            'Intended Audience :: Developers',
            'Intended Audience :: Education',
            'Intended Audience :: Science/Research',
            'License :: OSI Approved :: Apache Software License',
            'Programming Language :: C++',
            'Programming Language :: Python :: 2.7',
            'Programming Language :: Python :: 3.5',
            'Programming Language :: Python :: 3.6',
            'Programming Language :: Python :: 3.7',
            'Programming Language :: Python :: 3.8',
        ],
753 754
    )

755
# As there are a lot of files in purelib which causes many logs,
756
# we don't print them on the screen, and you can open `setup.py.log`
757
# for the full logs.
758 759
if os.path.exists('${SETUP_LOG_FILE}'):
    os.system('grep -v "purelib" ${SETUP_LOG_FILE}')