Now compiling ATIDLAS

python/setup.py

@@ -0,0 +1,91 @@
+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()