setup.py.in 1.7 KB
Newer Older
Z
zhangjinchao01 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This file is used to build paddle python binding package.
# It will be invoked by Makefile that generated by COMAKE
from setuptools import setup, Extension
import numpy as np
import api.paddle_ld_flags

# The extra links will passed from COMAKE
#   because generate paddle LDFLAGS is too complicated to do in setup.py
#   it just read COMAKE generated LDFLAGS.
extra_links = []
ldflags = api.paddle_ld_flags.PaddleLDFlag()
ldflags = ldflags.ldflag_str()
if ldflags is not None:
  extra_links.extend(ldflags.split(" "))

try:
  with open('.py_paddle_extra_link_flags', 'r') as f:
    for line in f:
      extra_links += line.split()
except:
  pass

setup(name="py_paddle",
L
liuyuan04 已提交
38
  version="@PADDLE_VERSION@",
Z
zhangjinchao01 已提交
39 40 41
  ext_modules=[
    Extension('py_paddle._swig_paddle',      # Build SWIG Extension.
      ['Paddle_wrap.cxx'],
L
liuyuan04 已提交
42
      extra_link_args=["-Xlinker", '-start-group'] +
Z
zhangjinchao01 已提交
43 44 45 46 47 48
                        extra_links + ["-Xlinker", "-end-group"]
    )
  ],
  packages=['py_paddle'],
  include_dirs = [np.get_include(), "../"],   # include numpy and paddle.
  install_requires = [
49
    'numpy>=1.8.0',      # The numpy is required.
Z
zhangjinchao01 已提交
50
    'protobuf>=2.4.1' # The paddle protobuf version
51 52
  ],
)