Commit 29f52eaa authored by Haotian Tang's avatar Haotian Tang
Browse files

[Minor] Remove a compilation argument.

parent 80222c63
......@@ -3,7 +3,7 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtensio
extra_compile_args = {
"cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"],
"nvcc": ["-O3", "-std=c++17", "-keep"],
"nvcc": ["-O3", "-std=c++17"],
}
setup(
......@@ -18,4 +18,4 @@ setup(
],
cmdclass={"build_ext": BuildExtension},
install_requires=["torch"],
)
\ No newline at end of file
)
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