本文整理汇总了Python中setuptools.Extension.extra_compile_args['nvcc']方法的典型用法代码示例。如果您正苦于以下问题：Python Extension.extra_compile_args['nvcc']方法的具体用法？Python Extension.extra_compile_args['nvcc']怎么用？Python Extension.extra_compile_args['nvcc']使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类setuptools.Extension的用法示例。
在下文中一共展示了Extension.extra_compile_args['nvcc']方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: len
# 需要导入模块: from setuptools import Extension [as 别名]
# 或者: from setuptools.Extension import extra_compile_args['nvcc'] [as 别名]
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',
author="Peter Wittek, Luca Calderaro",
author_email='[email protected]',
url="http://trotter-suzuki-mpi.github.io/",
platforms=["unix", "windows"],
description="Massively Parallel Trotter-Suzuki Solver",
ext_modules=[ts_module],
packages=["trottersuzuki"],
install_requires=['numpy'],