r/stylegan3
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Update custom ops to work with fresh install of VS 2022

Browse source 
for #115
This commit is contained in:
Janne Hellsten 2022-04-11 18:27:01 +03:00
parent a5a69f5829
commit 69c7ef0fbd
4 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
torch_utils/custom_ops.py
View file

@ -28,10 +28,10 @@ verbosity = 'brief' # Verbosity level: 'none', 'brief', 'full'
def _find_compiler_bindir(): def _find_compiler_bindir():
patterns = [ patterns = [
'C:/Program Files (x86)/Microsoft Visual Studio/*/Professional/VC/Tools/MSVC/*/bin/Hostx64/x64', 'C:/Program Files*/Microsoft Visual Studio/*/Professional/VC/Tools/MSVC/*/bin/Hostx64/x64',
'C:/Program Files (x86)/Microsoft Visual Studio/*/BuildTools/VC/Tools/MSVC/*/bin/Hostx64/x64', 'C:/Program Files*/Microsoft Visual Studio/*/BuildTools/VC/Tools/MSVC/*/bin/Hostx64/x64',
'C:/Program Files (x86)/Microsoft Visual Studio/*/Community/VC/Tools/MSVC/*/bin/Hostx64/x64', 'C:/Program Files*/Microsoft Visual Studio/*/Community/VC/Tools/MSVC/*/bin/Hostx64/x64',
'C:/Program Files (x86)/Microsoft Visual Studio */vc/bin', 'C:/Program Files*/Microsoft Visual Studio */vc/bin',
] ]
for pattern in patterns: for pattern in patterns:
matches = sorted(glob.glob(pattern)) matches = sorted(glob.glob(pattern))

2
torch_utils/ops/bias_act.py
View file

@ -43,7 +43,7 @@ def _init():
sources=['bias_act.cpp', 'bias_act.cu'], sources=['bias_act.cpp', 'bias_act.cu'],
headers=['bias_act.h'], headers=['bias_act.h'],
source_dir=os.path.dirname(__file__), source_dir=os.path.dirname(__file__),
extra_cuda_cflags=['--use_fast_math'], extra_cuda_cflags=['--use_fast_math', '--allow-unsupported-compiler'],
) )
return True return True

2
torch_utils/ops/filtered_lrelu.py
View file

@ -28,7 +28,7 @@ def _init():
sources=['filtered_lrelu.cpp', 'filtered_lrelu_wr.cu', 'filtered_lrelu_rd.cu', 'filtered_lrelu_ns.cu'], sources=['filtered_lrelu.cpp', 'filtered_lrelu_wr.cu', 'filtered_lrelu_rd.cu', 'filtered_lrelu_ns.cu'],
headers=['filtered_lrelu.h', 'filtered_lrelu.cu'], headers=['filtered_lrelu.h', 'filtered_lrelu.cu'],
source_dir=os.path.dirname(__file__), source_dir=os.path.dirname(__file__),
extra_cuda_cflags=['--use_fast_math'], extra_cuda_cflags=['--use_fast_math', '--allow-unsupported-compiler'],
) )
return True return True

2
torch_utils/ops/upfirdn2d.py
View file

@ -28,7 +28,7 @@ def _init():
sources=['upfirdn2d.cpp', 'upfirdn2d.cu'], sources=['upfirdn2d.cpp', 'upfirdn2d.cu'],
headers=['upfirdn2d.h'], headers=['upfirdn2d.h'],
source_dir=os.path.dirname(__file__), source_dir=os.path.dirname(__file__),
extra_cuda_cflags=['--use_fast_math'], extra_cuda_cflags=['--use_fast_math', '--allow-unsupported-compiler'],
) )
return True return True