Skip to content

GitLab

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

[Minor] Remove a compilation argument.

parent 80222c63
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
awq/kernels/setup.py
View file @ 29f52eaa
...@@ -3,7 +3,7 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtensio ...@@ -3,7 +3,7 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtensio
extra_compile_args = { extra_compile_args = {
"cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"], "cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"],
"nvcc": ["-O3", "-std=c++17", "-keep"], "nvcc": ["-O3", "-std=c++17"],
} }
setup( setup(
...@@ -18,4 +18,4 @@ setup( ...@@ -18,4 +18,4 @@ setup(
], ],
cmdclass={"build_ext": BuildExtension}, cmdclass={"build_ext": BuildExtension},
install_requires=["torch"], 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