|
|
|
@ -5,13 +5,13 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
|
|
|
|
_src_path = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
|
|
|
|
nvcc_flags = [
|
|
|
|
|
'-O3', '-std=c++14',
|
|
|
|
|
'-O3', '-std=c++17',
|
|
|
|
|
# '-lineinfo', # to debug illegal memory access
|
|
|
|
|
'-U__CUDA_NO_HALF_OPERATORS__', '-U__CUDA_NO_HALF_CONVERSIONS__', '-U__CUDA_NO_HALF2_OPERATORS__',
|
|
|
|
|
'-U__CUDA_NO_HALF_OPERATORS__', '-U__CUDA_NO_HALF_CONVERSIONS__', '-U__CUDA_NO_HALF2_OPERATORS__','-allow-unsupported-compiler',
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
if os.name == "posix":
|
|
|
|
|
c_flags = ['-O3', '-std=c++14']
|
|
|
|
|
c_flags = ['-O3', '-std=c++17']
|
|
|
|
|
elif os.name == "nt":
|
|
|
|
|
c_flags = ['/O2', '/std:c++17']
|
|
|
|
|
|
|
|
|
|