'trottersuzuki/src/cpublock.cpp',
'trottersuzuki/src/model.cpp',
'trottersuzuki/src/solver.cpp',
'trottersuzuki/trottersuzuki_wrap.cxx']
ts_module = Extension('_trottersuzuki', sources=sources_files,
include_dirs=[numpy_include, 'src'],
extra_compile_args=extra_compile_args,
libraries=libraries,
)
if CUDA is not None:
ts_module.sources += ['trottersuzuki/src/cc2kernel.cu',
'trottersuzuki/src/hybrid.cu']
ts_module.define_macros = [('CUDA', None)]
ts_module.include_dirs.append(CUDA['include'])
ts_module.library_dirs = [CUDA['lib']]
ts_module.libraries += ['cudart', 'cublas']
ts_module.runtime_library_dirs = [CUDA['lib']]
if len(ts_module.extra_compile_args['cc']) > 0:
extra_args = ts_module.extra_compile_args['cc'][0]
else:
extra_args = ""
ts_module.extra_compile_args['nvcc'] = ['-use_fast_math',
'--ptxas-options=-v', '-c',
'--compiler-options',
'-fPIC ' + extra_args]
cmdclass = {'build_ext': custom_build_ext}
setup(name='trottersuzuki',
version='1.5.2',
if 'CC' in os.environ and 'clang-omp' in os.environ['CC']:
openmp = 'iomp5'
else:
openmp = 'gomp'
cmdclass = {'build_ext': custom_build_ext}
somoclu_module = Extension('_somoclu_wrap',
sources=sources_files,
include_dirs=[numpy_include, 'src'],
extra_compile_args={'cc': extra_compile_args},
libraries=[openmp],
)
if CUDA is not None:
somoclu_module.sources.append('somoclu/src/denseGpuKernels.cu')
somoclu_module.define_macros = [('CUDA', None)]
somoclu_module.include_dirs.append(CUDA['include'])
somoclu_module.library_dirs = [CUDA['lib']]
somoclu_module.libraries += ['cudart', 'cublas']
somoclu_module.runtime_library_dirs = [CUDA['lib']]
somoclu_module.extra_compile_args['nvcc']=['-use_fast_math',
'--ptxas-options=-v', '-c',
'--compiler-options','-fPIC ' +
extra_compile_args[0]]
try:
setup(name='somoclu',
version='1.6.1',
license='GPL3',
author="Peter Wittek, Shi Chao Gao",
author_email="",
if 'CC' in os.environ and 'clang-omp' in os.environ['CC']:
openmp = 'iomp5'
else:
openmp = 'gomp'
cmdclass = {'build_ext': custom_build_ext}
somoclu_module = Extension('_somoclu_wrap',
sources=sources_files,
include_dirs=[numpy_include, 'src'],
extra_compile_args={'cc': extra_compile_args},
libraries=[openmp],
)
if CUDA is not None:
somoclu_module.sources.append('somoclu/src/denseGpuKernels.cu')
somoclu_module.define_macros = [('CUDA', None)]
somoclu_module.include_dirs.append(CUDA['include'])
somoclu_module.library_dirs = [CUDA['lib']]
somoclu_module.libraries += ['cudart', 'cublas']
somoclu_module.runtime_library_dirs = [CUDA['lib']]
somoclu_module.extra_compile_args['nvcc']=['-use_fast_math',
'--ptxas-options=-v', '-c',
'--compiler-options','-fPIC ' +
extra_compile_args[0]]
try:
setup(name='somoclu',
version='1.7.3',
license='GPL3',
author="Peter Wittek, Shi Chao Gao",
author_email="",
]
ts_module = Extension(
'_trottersuzuki',
sources=sources_files,
include_dirs=[numpy_include, 'src'],
extra_compile_args=extra_compile_args,
libraries=libraries,
)
if CUDA is not None:
ts_module.sources += [
'trottersuzuki/src/cc2kernel.cu', 'trottersuzuki/src/hybrid.cu'
]
ts_module.define_macros = [('CUDA', None)]
ts_module.include_dirs.append(CUDA['include'])
ts_module.library_dirs = [CUDA['lib']]
ts_module.libraries += ['cudart', 'cublas']
ts_module.runtime_library_dirs = [CUDA['lib']]
if len(ts_module.extra_compile_args['cc']) > 0:
extra_args = ts_module.extra_compile_args['cc'][0]
else:
extra_args = ""
ts_module.extra_compile_args['nvcc'] = [
'-use_fast_math', '--ptxas-options=-v', '-c', '--compiler-options',
'-fPIC ' + extra_args
]
cmdclass = {'build_ext': custom_build_ext}
setup(name='trottersuzuki',
version='1.5.2',
license='GPL3',
openmp = "iomp5"
else:
openmp = "gomp"
cmdclass = {"build_ext": custom_build_ext}
somoclu_module = Extension(
"somoclu._somoclu_wrap",
sources=sources_files,
include_dirs=[numpy_include, "src"],
extra_compile_args={"cc": extra_compile_args},
libraries=[openmp],
)
if CUDA is not None:
somoclu_module.sources.append("somoclu/src/denseGpuKernels.cu")
somoclu_module.define_macros = [("CUDA", None)]
somoclu_module.include_dirs.append(CUDA["include"])
somoclu_module.library_dirs = [CUDA["lib"]]
somoclu_module.libraries += ["cudart", "cublas"]
somoclu_module.runtime_library_dirs = [CUDA["lib"]]
somoclu_module.extra_compile_args["nvcc"] = [
"-use_fast_math",
"--ptxas-options=-v",
"-c",
"--compiler-options",
"-fPIC " + extra_compile_args[0],
]
try:
setup(
name="somoclu",
version="1.7.5",
license="GPL3",
