forked from facebookresearch/Hanabi_SPARTA
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
42 lines (37 loc) · 1.29 KB
/
setup.py
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
38
39
40
41
42
#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CppExtension
import sys
import os
OPTIONAL_SRC = []
if int(os.environ.get("INSTALL_TORCHBOT", 0)):
OPTIONAL_SRC = ["csrc/TorchBot.cc"]
boost_libs = ["boost_fiber", "boost_thread", "boost_context"]
if sys.platform == "darwin":
boost_libs = [lib + '-mt' for lib in boost_libs]
setup(
name='hanabi_lib',
ext_modules=[
CppExtension('hanabi_lib', [
"csrc/extension.cc",
"csrc/SimpleBot.cc",
"csrc/HolmesBot.cc",
"csrc/SmartBot.cc",
"csrc/SearchBot.cc",
"csrc/JointSearchBot.cc",
"csrc/HanabiServer.cc",
"csrc/BotUtils.cc",
] + OPTIONAL_SRC,
extra_compile_args=['-fPIC', '-std=c++1y', '-Wno-deprecated', '-O3', '-Wno-sign-compare', '-D_GLIBCXX_USE_CXX11_ABI=0', '-DCARD_ID=1'],
libraries = ['z'] + boost_libs,
library_dirs=['/usr/local/lib'],
include_dirs=['csrc'],
undef_macros=['NDEBUG'])
],
cmdclass={"build_ext": BuildExtension},
)