setup.py.in 28.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
from setuptools import Command
from setuptools import setup, Distribution, Extension
from setuptools.command.install import install as InstallCommandBase
16
from setuptools.command.egg_info import egg_info
17 18


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

Y
Yancey 已提交
23 24
RC      = 0

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

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

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

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

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

47
    return 0
48

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

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

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

58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
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'

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

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

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

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

Y
Yancey 已提交
106
def show():
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 147
    """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 已提交
148
    if istaged:
149 150 151 152 153
        print('full_version:', full_version)
        print('major:', major)
        print('minor:', minor)
        print('patch:', patch)
        print('rc:', rc)
Y
Yancey 已提交
154
    else:
155
        print('commit:', commit)
156 157
    print('cuda:', cuda_version)
    print('cudnn:', cudnn_version)
L
Luo Tao 已提交
158 159 160

def mkl():
    return with_mkl
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 194

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 已提交
195 196
'''
    commit = git_commit()
197 198 199 200 201 202 203 204 205

    dirname = os.path.dirname(filename)

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

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

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

221 222 223 224 225 226 227 228 229 230 231 232 233 234
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 已提交
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 262
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 已提交
263

264

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

403 404
with open('@PADDLE_SOURCE_DIR@/python/requirements.txt') as f:
    setup_requires = f.read().splitlines()
405

406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429
# 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

430
# the prefix is sys.prefix which should always be usr
L
Luo Tao 已提交
431
paddle_bins = ''
432

T
Tao Luo 已提交
433 434
if not '${WIN32}':
    paddle_bins = ['${PADDLE_BINARY_DIR}/paddle/scripts/paddle']
435 436 437 438 439 440

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 已提交
441
package_data['paddle.fluid'] += ['${PADDLE_BINARY_DIR}/python/paddle/cost_model/static_op_benchmark.json']
442
if '${HAS_NOAVX_CORE}' == 'ON':
443
    package_data['paddle.fluid'] += ['core_noavx' + ('.so' if os.name != 'nt' else '.pyd')]
P
peizhilin 已提交
444

L
Luo Tao 已提交
445
package_dir={
446
    '': '${PADDLE_BINARY_DIR}/python',
L
Luo Tao 已提交
447 448 449 450
    # 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 已提交
451
    'paddle.fluid': '${PADDLE_BINARY_DIR}/python/paddle/fluid',
L
Luo Tao 已提交
452
}
453

454 455
# put all thirdparty libraries in paddle.libs
libs_path='${PADDLE_BINARY_DIR}/python/paddle/libs'
P
peizhilin 已提交
456

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

461 462 463 464 465 466 467 468 469 470 471 472 473
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)

474
if '${WITH_MKL}' == 'ON':
P
peizhilin 已提交
475 476 477
    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 已提交
478 479
else:
    if os.name == 'nt':
P
peizhilin 已提交
480
        # copy the openblas.dll
481
        shutil.copy('${OPENBLAS_SHARED_LIB}', libs_path)
P
peizhilin 已提交
482
        package_data['paddle.libs'] += ['openblas' + ext_name]
W
Wilber 已提交
483
    elif os.name == 'posix' and platform.machine() == 'aarch64' and '${OPENBLAS_LIB}'.endswith('so'):
H
houj04 已提交
484 485
        # copy the libopenblas.so on linux+aarch64
        # special: core_noavx.so depends on 'libopenblas.so.0', not 'libopenblas.so'
W
Wilber 已提交
486 487 488
        if os.path.exists('${OPENBLAS_LIB}' + '.0'):
            shutil.copy('${OPENBLAS_LIB}' + '.0', libs_path)
            package_data['paddle.libs'] += ['libopenblas.so.0']
P
peizhilin 已提交
489

490 491 492
if '${WITH_LITE}' == 'ON':
    shutil.copy('${LITE_SHARED_LIB}', libs_path)
    package_data['paddle.libs']+=['libpaddle_full_api_shared' + ext_name]
493 494 495 496 497 498
    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]
499

500 501
if '${WITH_CINN}' == 'ON':
    shutil.copy('${CINN_LIB_LOCATION}/${CINN_LIB_NAME}', libs_path)
502
    shutil.copy('${CINN_INCLUDE_DIR}/cinn/runtime/cuda/cinn_cuda_runtime_source.cuh', libs_path)
503
    package_data['paddle.libs']+=['libcinnapi.so']
504
    package_data['paddle.libs']+=['cinn_cuda_runtime_source.cuh']
505

X
xujiaqi01 已提交
506 507
if '${WITH_PSLIB}' == 'ON':
    shutil.copy('${PSLIB_LIB}', libs_path)
X
xujiaqi01 已提交
508 509
    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 已提交
510 511
    package_data['paddle.libs'] += ['libps' + ext_name]

S
Sang Ik Lee 已提交
512
if '${WITH_MKLDNN}' == 'ON':
P
peizhilin 已提交
513
    if '${CMAKE_BUILD_TYPE}' == 'Release' and os.name != 'nt':
P
peizhilin 已提交
514 515 516 517
        # 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 已提交
518
        # change rpath of libdnnl.so.1, add $ORIGIN/ to it.
P
peizhilin 已提交
519
        # The reason is that all thirdparty libraries in the same directory,
A
Adam 已提交
520
        # thus, libdnnl.so.1 will find libmklml_intel.so and libiomp5.so.
P
peizhilin 已提交
521 522
        command = "patchelf --set-rpath '$ORIGIN/' ${MKLDNN_SHARED_LIB}"
        if os.system(command) != 0:
A
Adam 已提交
523
            raise Exception("patch libdnnl.so failed, command: %s" % command)
S
Sang Ik Lee 已提交
524
    shutil.copy('${MKLDNN_SHARED_LIB}', libs_path)
A
Adam 已提交
525 526
    if os.name != 'nt':
        shutil.copy('${MKLDNN_SHARED_LIB_1}', libs_path)
527 528
        shutil.copy('${MKLDNN_SHARED_LIB_2}', libs_path)
        package_data['paddle.libs']+=['libmkldnn.so.0', 'libdnnl.so.1', 'libdnnl.so.2']
A
Adam 已提交
529 530 531
    else:
        package_data['paddle.libs']+=['mkldnn.dll']

532 533 534 535 536 537 538 539 540 541 542 543
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']

544 545 546 547 548 549 550 551 552 553 554 555 556
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}',
557
                                  '${XPU_RT_LIB_NAME}']
558

559 560 561 562
if '${WITH_XPU_BKCL}' == 'ON':
    shutil.copy('${XPU_BKCL_LIB}', libs_path)
    package_data['paddle.libs']+=['${XPU_BKCL_LIB_NAME}']

563
# remove unused paddle/libs/__init__.py
P
peizhilin 已提交
564 565
if os.path.isfile(libs_path+'/__init__.py'):
    os.remove(libs_path+'/__init__.py')
566 567
package_dir['paddle.libs']=libs_path

568

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

P
peizhilin 已提交
587
ext_modules = [Extension('_foo', ['stub.cc'])]
P
peizhilin 已提交
588 589 590 591 592 593
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 已提交
594
    ext_modules = []
595 596
elif sys.platform == 'darwin':
    ext_modules = []
T
tensor-tang 已提交
597

598
def find_files(pattern, root, recursive=False):
599
    for dirpath, _, files in os.walk(root):
600 601 602 603
        for filename in fnmatch.filter(files, pattern):
            yield os.path.join(dirpath, filename)
        if not recursive:
            break
604 605

headers = (
606
    # paddle level api headers
607
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle')) +
608
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/api')) +  # phi unify api header
609
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/api/ext')) +  # custom op api
610 611 612
    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)
613 614
    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
615
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends', recursive=True)) +  # phi backends headers
616 617 618
    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
619
    # utila api headers
620
    list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/utils', recursive=True)))  # paddle utils headers
621

622 623 624
if '${WITH_MKLDNN}' == 'ON':
    headers += list(find_files('*', '${MKLDNN_INSTALL_DIR}/include')) # mkldnn

625
if '${WITH_GPU}' == 'ON' or '${WITH_ROCM}' == 'ON':
626 627
    # externalErrorMsg.pb for External Error message
    headers += list(find_files('*.pb', '${externalError_INCLUDE_DIR}'))
628 629 630 631 632

class InstallCommand(InstallCommandBase):
    def finalize_options(self):
        ret = InstallCommandBase.finalize_options(self)
        self.install_lib = self.install_platlib
633
        self.install_headers = os.path.join(self.install_platlib, 'paddle', 'include')
634
        return ret
635

636 637 638 639
class InstallHeaders(Command):
    """Override how headers are copied.
    """
    description = 'install C/C++ header files'
640

641 642 643 644 645
    user_options = [('install-dir=', 'd',
                     'directory to install header files to'),
                    ('force', 'f',
                     'force installation (overwrite existing files)'),
                   ]
646

647
    boolean_options = ['force']
648

649 650 651 652
    def initialize_options(self):
        self.install_dir = None
        self.force = 0
        self.outfiles = []
653

654 655 656 657
    def finalize_options(self):
        self.set_undefined_options('install',
                                   ('install_headers', 'install_dir'),
                                   ('force', 'force'))
658

659 660 661 662
    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:
663
            # paddle headers
664 665 666 667
            install_dir = re.sub('@PADDLE_SOURCE_DIR@/', '', header)
        else:
            # third_party
            install_dir = re.sub('${THIRD_PARTY_PATH}', 'third_party', header)
668
            patterns = ['install/mkldnn/include']
669 670
            for pattern in patterns:
                install_dir = re.sub(pattern, '', install_dir)
671 672 673 674
        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)
675

676 677 678 679 680 681 682 683
    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)
684

685 686
    def get_inputs(self):
        return self.distribution.headers or []
687

688 689 690
    def get_outputs(self):
        return self.outfiles

691 692 693 694 695 696 697 698 699 700 701
class EggInfo(egg_info):
    """Copy license file into `.dist-info` folder."""

    def run(self):
        # don't duplicate license into `.dist-info` when building a distribution
        if not self.distribution.have_run.get('install', True):
            self.mkpath(self.egg_info)
            self.copy_file("@PADDLE_SOURCE_DIR@/LICENSE", self.egg_info)

        egg_info.run(self)

702 703 704 705
# we redirect setuptools log for non-windows
if sys.platform != 'win32':
    @contextmanager
    def redirect_stdout():
706 707 708 709 710 711 712
        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()
713 714 715 716 717
else:
    @contextmanager
    def redirect_stdout():
        yield

718 719
# Log for PYPI
if sys.version_info > (3,0):
720
    with open("@PADDLE_BINARY_DIR@/python/paddle/README.rst", "r", encoding='UTF-8') as f:
721 722
        long_description = f.read()
else:
723
    with open("@PADDLE_BINARY_DIR@/python/paddle/README.rst", "r")as f:
724 725
        long_description = unicode(f.read(), 'UTF-8')

W
wuhuanzhou 已提交
726 727 728 729 730 731
# 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)

732 733 734 735
with redirect_stdout():
    setup(name='${PACKAGE_NAME}',
        version='${PADDLE_VERSION}',
        description='Parallel Distributed Deep Learning',
736 737 738 739 740 741 742 743 744 745
        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',
746
        packages=packages,
747
        install_requires=setup_requires,
748 749 750 751 752 753 754 755 756
        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,
757
            'egg_info': EggInfo,
758 759 760
        },
        entry_points={
            'console_scripts': [
761
                'fleetrun = paddle.distributed.launch.main:launch'
762
            ]
763 764 765 766 767 768 769 770 771 772 773 774 775 776 777
        },
        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',
        ],
778 779
    )

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