File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 88from torch .utils .cpp_extension import CppExtension , CUDAExtension , CUDA_HOME
99
1010
11- # Windows users: Edit both of these to contain your Visual Studio include path, i.e.
11+ # Windows users: Edit both of these to contain your VS include path, i.e.
1212# cxx_extra_compile_args = ['-I{VISUAL_STUDIO_DIR}\\include']
13- # nvcc_extra_compile_args = ['-arch=sm_35', '--expt-relaxed-constexpr' , '-I{VISUAL_STUDIO_DIR}\\include']
13+ # nvcc_extra_compile_args = [... , '-I{VISUAL_STUDIO_DIR}\\include']
1414cxx_extra_compile_args = []
1515nvcc_extra_compile_args = ['-arch=sm_35' , '--expt-relaxed-constexpr' ]
1616
17- # Windows users: Edit both of these to contain your Visual Studio library path, i.e.
17+ # Windows users: Edit both of these to contain your VS library path, i.e.
1818# cxx_extra_link_args = ['/LIBPATH:{VISUAL_STUDIO_DIR}\\lib\\{x86|x64}']
1919# nvcc_extra_link_args = ['/LIBPATH:{VISUAL_STUDIO_DIR}\\lib\\{x86|x64}']
2020cxx_extra_link_args = []
You can’t perform that action at this time.
0 commit comments