setup.py.in 27.7 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}']

A
Allen Guo 已提交
550 551 552 553
if '${WITH_IPU}' == 'ON':
    shutil.copy('${PADDLE_IPU_LIB}', libs_path)
    package_data['paddle.libs'] += ['libpaddle_ipu' + ext_name]

554
# remove unused paddle/libs/__init__.py
P
peizhilin 已提交
555 556
if os.path.isfile(libs_path+'/__init__.py'):
    os.remove(libs_path+'/__init__.py')
557 558
package_dir['paddle.libs']=libs_path

559

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

P
peizhilin 已提交
578
ext_modules = [Extension('_foo', ['stub.cc'])]
P
peizhilin 已提交
579 580 581 582 583 584
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 已提交
585
    ext_modules = []
586 587
elif sys.platform == 'darwin':
    ext_modules = []
T
tensor-tang 已提交
588

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

headers = (
597
    # paddle level api headers
598
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle')) +
599
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/api')) +  # phi unify api header
600
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/api/ext')) +  # custom op api
601 602 603
    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)
604 605
    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
606
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends', recursive=True)) +  # phi backends headers
607 608 609
    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
610
    # utila api headers
611
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/utils', recursive=True)))  # paddle utils headers
612

613 614 615
if '${WITH_MKLDNN}' == 'ON':
    headers += list(find_files('*', '${MKLDNN_INSTALL_DIR}/include')) # mkldnn

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

class InstallCommand(InstallCommandBase):
    def finalize_options(self):
        ret = InstallCommandBase.finalize_options(self)
        self.install_lib = self.install_platlib
624
        self.install_headers = os.path.join(self.install_platlib, 'paddle', 'include')
625
        return ret
626

627 628 629 630
class InstallHeaders(Command):
    """Override how headers are copied.
    """
    description = 'install C/C++ header files'
631

632 633 634 635 636
    user_options = [('install-dir=', 'd',
                     'directory to install header files to'),
                    ('force', 'f',
                     'force installation (overwrite existing files)'),
                   ]
637

638
    boolean_options = ['force']
639

640 641 642 643
    def initialize_options(self):
        self.install_dir = None
        self.force = 0
        self.outfiles = []
644

645 646 647 648
    def finalize_options(self):
        self.set_undefined_options('install',
                                   ('install_headers', 'install_dir'),
                                   ('force', 'force'))
649

650 651 652 653
    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:
654
            # paddle headers
655 656 657 658
            install_dir = re.sub('@PADDLE_SOURCE_DIR@/', '', header)
        else:
            # third_party
            install_dir = re.sub('${THIRD_PARTY_PATH}', 'third_party', header)
659
            patterns = ['install/mkldnn/include']
660 661
            for pattern in patterns:
                install_dir = re.sub(pattern, '', install_dir)
662 663 664 665
        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)
666

667 668 669 670 671 672 673 674
    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)
675

676 677
    def get_inputs(self):
        return self.distribution.headers or []
678

679 680 681
    def get_outputs(self):
        return self.outfiles

682 683 684 685
# we redirect setuptools log for non-windows
if sys.platform != 'win32':
    @contextmanager
    def redirect_stdout():
686 687 688 689 690 691 692
        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()
693 694 695 696 697
else:
    @contextmanager
    def redirect_stdout():
        yield

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

W
wuhuanzhou 已提交
706 707 708 709 710 711
# 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)

712 713 714 715
with redirect_stdout():
    setup(name='${PACKAGE_NAME}',
        version='${PADDLE_VERSION}',
        description='Parallel Distributed Deep Learning',
716 717 718 719 720 721 722 723 724 725
        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',
726
        packages=packages,
727
        install_requires=setup_requires,
728 729 730 731 732 733 734 735 736
        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,
737 738 739
        },
        entry_points={
            'console_scripts': [
740
                'fleetrun = paddle.distributed.launch.main:launch'
741
            ]
742 743 744 745 746 747 748 749 750 751 752 753 754 755 756
        },
        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',
        ],
757 758
    )

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