Unverified Commit 0b354c1d authored by Richard Xue's avatar Richard Xue Committed by GitHub
Browse files

Update setup.py (#529)

parent 0f8eb38f
......@@ -60,7 +60,7 @@ def cuda_extension():
pkg='torchani.cuaev._real_cuaev',
sources=['torchani/cuaev/aev.cu'],
include_dirs=maybe_download_cub(),
extra_compile_args={'nvcc': nvcc_args},
extra_compile_args={'cxx': ['-std=c++14'], 'nvcc': nvcc_args},
optional=True)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment