Esempio n. 1
0
install_requires = []

cuda_version = ''.join(
    os.environ.get('CUDA_VERSION', 'unknown').split('.')[:2])
name = 'nvstrings-cuda{}'.format(cuda_version)
version = os.environ.get('GIT_DESCRIBE_TAG', '0.0.0.dev0').lstrip('v')

setup(name=name,
      description='CUDA strings Python bindings',
      url='https://github.com/rapidsai/custrings',
      version=version,
      classifiers=[
          "Intended Audience :: Developers", "Programming Language :: Python",
          "Programming Language :: Python :: 3.6",
          "Programming Language :: Python :: 3.7"
      ],
      py_modules=['nvstrings', 'nvcategory', 'nvtext'],
      author='NVIDIA Corporation',
      license="Apache",
      install_requires=install_requires,
      ext_modules=[
          CMakeExtension('pyniNVStrings', 'cpp'),
          CMakeExtension('pyniNVCategory', 'cpp'),
          CMakeExtension('pyniNVText', 'cpp')
      ],
      cmdclass={'build_ext': CMakeBuildExt},
      zip_safe=False)

convert_to_manylinux(name, version)
Esempio n. 2
0
        'lib64': os.path.join(home, 'lib64')
    }
    for k, v in iter(cudaconfig.items()):
        if not os.path.exists(v):
            raise EnvironmentError(
                'The CUDA %s path could not be located in %s' % (k, v))

    return cudaconfig


cuda_include = locate_cuda()['include']

cython_files = ['python/cugraph/*.pyx']

extensions = [
    CMakeExtension('cugraph', sourcedir='cpp'),
    Extension("cugraph",
              sources=cython_files,
              include_dirs=[
                  numpy_include, cudf_include, cuda_include, 'cpp/src',
                  'cpp/include', '../gunrock',
                  '../gunrock/externals/moderngpu/include',
                  '../gunrock/externals/cub'
              ],
              library_dirs=[get_python_lib(),
                            distutils_dir_name('lib')],
              libraries=['nvgraph'],
              language='c++',
              extra_compile_args=['-std=c++11'])
]
Esempio n. 3
0
from setuptools import setup
from cmake_setuptools import CMakeExtension, CMakeBuildExt

setup(name="cppderiv",
      description="Compiled assortment of numerical methods to solve ODEs",
      version="0.1.0beta",
      ext_modules=[CMakeExtension("all")],
      cmdclass={"build_ext": CMakeBuildExt})
Esempio n. 4
0
    'numpy', 'cython>=0.29<0.30',
    'cudf-cuda{}>={},<{}'.format(cuda_version, cudf_version, max_cudf_version)
]

exc_list = [
    'python/cuml/linear_model/linear_regression_mg.pyx',
    'python/cuml/decomposition/tsvd_mg.pyx'
]

cython_files = ['python/cuml/*/*.pyx']

cuda_include_dir = '/usr/local/cuda/include'
cuda_lib_dir = "/usr/local/cuda/lib"

extensions = [
    CMakeExtension('cuml', 'cuML'),
    Extension("*",
              sources=cython_files,
              include_dirs=[
                  'cuML/src', 'cuML/external', 'cuML/external/ml-prims/src',
                  'cuML/external/ml-prims/external/cutlass',
                  'cuML/external/cutlass',
                  'cuML/external/ml-prims/external/cub', cuda_include_dir
              ],
              library_dirs=[get_python_lib(),
                            distutils_dir_name('lib')],
              libraries=['cuml'],
              language='c++',
              runtime_library_dirs=['$ORIGIN', cuda_lib_dir],
              extra_compile_args=['-std=c++11'])
]
Esempio n. 5
0
cuda_version = "".join(
    os.environ.get("CUDA_VERSION", "unknown").split(".")[:2]
)
name = "nvstrings-cuda{}".format(cuda_version)
version = os.environ.get("GIT_DESCRIBE_TAG", "0.0.0.dev0").lstrip("v")

setup(
    name=name,
    description="CUDA strings Python bindings",
    url="https://github.com/rapidsai/cudf",
    version=version,
    classifiers=[
        "Intended Audience :: Developers",
        "Programming Language :: Python",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
    ],
    py_modules=["nvstrings", "nvcategory", "nvtext"],
    author="NVIDIA Corporation",
    license="Apache",
    install_requires=install_requires,
    ext_modules=[
        CMakeExtension("pyniNVStrings", "cpp"),
        CMakeExtension("pyniNVCategory", "cpp"),
        CMakeExtension("pyniNVText", "cpp"),
    ],
    cmdclass={"build_ext": CMakeBuildExt},
    zip_safe=False,
)
Esempio n. 6
0
            self.outfiles.append(out)


setup(
    name='flatboobs',
    version=__version__,
    author='Ildar Akhmetgaleev',
    description='FlatBuffer reader/writer generator',
    long_description=long_description,
    license='MIT',
    packages=find_packages(exclude=['third_party', 'docs', 'tests*', 'utils']),
    zip_safe=False,
    include_package_data=True,
    ext_package='flatboobs',
    ext_modules=[
        CMakeExtension('idl'),
    ],
    cmdclass={
        'build_ext': CMakeBuildExt,
        'install_headers': install_flatboobs_headers,
    },
    # This part is good for when the setup.py itself cannot proceed until dependencies
    # in ``setup_requires`` are met. If you also need some/all of the dependencies in
    # ``setup_requires`` to run your module, be sure to have them in the install_requirements to.
    # setup_requires=[],
    #
    # You may specify additional packages for a more feature filled install.
    # Example of a extras_require where one has to do:
    #     python -m pip install module_template    (to get the default package)
    #     python -m pip install module_template[test]   (to get additional dependencies
    #                                                    to enable ``test`` functionality)
Esempio n. 7
0
from cmake_setuptools import CMakeExtension, CMakeBuildExt
from setuptools import setup

setup(
    name='lmbspecialops',
    description='A python wrapper for tf to ease creation of network definitions',
    version='A collection of tensorflow ops',
    url='https://github.com/lmb-freiburg/lmbspecialops',
    license='GPLv3.0',
    author='lmb-freiburg',
    author_email='',
    python_requires='>=3.5',
    setup_requires=['cmake-setuptools', 'tensorflow'],
    ext_modules=[CMakeExtension('lmbspecialops')],
    cmdclass={'build_ext': CMakeBuildExt},
    package_dir={'': 'python'},
    packages=['lmbspecialops'],
    zip_safe=False,
)