92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
|
import os
|
||
|
|
from distutils.command.build_ext import build_ext
|
||
|
|
from setuptools import Extension, setup
|
||
|
|
from distutils.sysconfig import get_python_inc
|
||
|
|
from distutils import sysconfig
|
||
|
|
|
||
|
|
platform_cflags = {}
|
||
|
|
platform_ldflags = {}
|
||
|
|
platform_libs = {}
|
||
|
|
class build_ext_subclass(build_ext):
|
||
|
|
"""Shamelessly stolen from
|
||
|
|
https://stackoverflow.com/questions/724664
|
||
|
|
"""
|
||
|
|
def build_extensions(self):
|
||
|
|
c = self.compiler.compiler_type
|
||
|
|
if c in platform_cflags.keys():
|
||
|
|
for e in self.extensions:
|
||
|
|
e.extra_compile_args = platform_cflags[c]
|
||
|
|
if c in platform_ldflags.keys():
|
||
|
|
for e in self.extensions:
|
||
|
|
e.extra_link_args = platform_ldflags[c]
|
||
|
|
if c in platform_libs.keys():
|
||
|
|
for e in self.extensions:
|
||
|
|
try:
|
||
|
|
e.libraries += platform_libs[c]
|
||
|
|
except:
|
||
|
|
e.libraries = platform_libs[c]
|
||
|
|
build_ext.build_extensions(self)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
|
||
|
|
def remove_prefixes(optlist, bad_prefixes):
|
||
|
|
for bad_prefix in bad_prefixes:
|
||
|
|
for i, flag in enumerate(optlist):
|
||
|
|
if flag.startswith(bad_prefix):
|
||
|
|
optlist.pop(i)
|
||
|
|
break
|
||
|
|
return optlist
|
||
|
|
|
||
|
|
cvars = sysconfig.get_config_vars()
|
||
|
|
cvars['OPT'] = str.join(' ', remove_prefixes(cvars['OPT'].split(), ['-g', '-O', '-Wstrict-prototypes', '-DNDEBUG']))
|
||
|
|
|
||
|
|
DEFINES = [('VIENNACL_WITH_OPENCL','1')]
|
||
|
|
INCLUDE_DIRS = [os.path.dirname(os.path.dirname(os.path.abspath(__file__)))]
|
||
|
|
|
||
|
|
setup(
|
||
|
|
name="atidlas",
|
||
|
|
version=[],
|
||
|
|
description="Auto-tuned input-dependent linear algebra subroutines",
|
||
|
|
author='Philippe Tillet',
|
||
|
|
author_email='ptillet@g.harvard.edu',
|
||
|
|
classifiers=[
|
||
|
|
'Environment :: Console',
|
||
|
|
'Development Status :: 5 - Production/Stable',
|
||
|
|
'Intended Audience :: Developers',
|
||
|
|
'Intended Audience :: Other Audience',
|
||
|
|
'Intended Audience :: Science/Research',
|
||
|
|
'License :: OSI Approved :: MIT License',
|
||
|
|
'Natural Language :: English',
|
||
|
|
'Programming Language :: C++',
|
||
|
|
'Programming Language :: Python',
|
||
|
|
'Programming Language :: Python :: 2',
|
||
|
|
'Programming Language :: Python :: 2.6',
|
||
|
|
'Programming Language :: Python :: 2.7',
|
||
|
|
'Programming Language :: Python :: 3',
|
||
|
|
'Programming Language :: Python :: 3.2',
|
||
|
|
'Programming Language :: Python :: 3.3',
|
||
|
|
'Programming Language :: Python :: 3.4',
|
||
|
|
'Topic :: Scientific/Engineering',
|
||
|
|
'Topic :: Scientific/Engineering :: Mathematics',
|
||
|
|
'Topic :: Scientific/Engineering :: Physics',
|
||
|
|
'Topic :: Scientific/Engineering :: Machine Learning',
|
||
|
|
],
|
||
|
|
|
||
|
|
packages=["atidlas"],
|
||
|
|
ext_package="atidlas",
|
||
|
|
ext_modules=[Extension(
|
||
|
|
'_atidlas',[os.path.join("src", "_atidlas.cpp")],
|
||
|
|
extra_compile_args= [],
|
||
|
|
extra_link_args=[],
|
||
|
|
define_macros=DEFINES,
|
||
|
|
undef_macros=[],
|
||
|
|
include_dirs=INCLUDE_DIRS,
|
||
|
|
library_dirs=[],
|
||
|
|
libraries=['OpenCL']
|
||
|
|
)],
|
||
|
|
cmdclass={'build_ext': build_ext_subclass}
|
||
|
|
)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|