OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Add lazy_init.h into build for Windows and refactor code (#5365)

Browse Source 
* Add lazy_init.h into build for Windows and refactor code

* Remove minor bugs
This commit is contained in:
peterjc123 2018-02-23 13:05:43 +08:00 committed by Soumith Chintala
parent 8904616028
commit 013ed5b88f
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
setup.py
View File

@ -405,13 +405,26 @@ class build_ext(build_ext_parent):
if WITH_CUDA: if WITH_CUDA:
THCUNN.extra_link_args += [_C_LIB] THCUNN.extra_link_args += [_C_LIB]
else: else:
# To generate .obj files for AutoGPU for the export class # To generate .obj files for those .h files for the export class
# a header file cannot build, so it has to be copied to someplace as a source file # a header file cannot build, so it has to be copied to someplace as a source file
if not os.path.exists("torch/csrc/generated"): temp_dir = 'torch/csrc/generated'
os.mkdir("torch/csrc/generated") hfile_list = ['torch/csrc/cuda/AutoGPU.h',
if os.path.exists("torch/csrc/generated/AutoGPU_cpu_win.cpp"): 'torch/csrc/cuda/lazy_init.h']
os.remove("torch/csrc/generated/AutoGPU_cpu_win.cpp") hname_list = [os.path.basename(hfile) for hfile in hfile_list]
shutil.copyfile("torch/csrc/cuda/AutoGPU.h", "torch/csrc/generated/AutoGPU_cpu_win.cpp") rname_list = [os.path.splitext(hname)[0]
for hname in hname_list]
cfile_list = [temp_dir + '/' + rname +
'_cpu_win.cpp' for rname in rname_list]
if not os.path.exists(temp_dir):
os.mkdir(temp_dir)
for hfile, cfile in zip(hfile_list, cfile_list):
if os.path.exists(cfile):
os.remove(cfile)
shutil.copyfile(hfile, cfile)
C.main_sources += cfile_list
if WITH_NINJA: if WITH_NINJA:
# before we start the normal build make sure all generated code # before we start the normal build make sure all generated code
# gets built # gets built
@ -638,9 +651,6 @@ if WITH_DISTRIBUTED:
include_dirs += [tmp_install_path + "/include/THD"] include_dirs += [tmp_install_path + "/include/THD"]
main_link_args += [THD_LIB] main_link_args += [THD_LIB]
if IS_WINDOWS and not WITH_CUDA:
main_sources += ["torch/csrc/generated/AutoGPU_cpu_win.cpp"]
if WITH_CUDA: if WITH_CUDA:
nvtoolext_lib_name = None nvtoolext_lib_name = None
if IS_WINDOWS: if IS_WINDOWS: