corrine add msvc path in sustom_ops.py

3 years ago · 6ad85bd366
parent 1a2d37d03d
commit 6ad85bd366
1 changed files with 1 additions and 0 deletions
--- a/torch_utils/custom_ops.py
+++ b/torch_utils/custom_ops.py
@ -32,6 +32,7 @@ def _find_compiler_bindir():
        'C:/Program Files*/Microsoft Visual Studio/*/BuildTools/VC/Tools/MSVC/*/bin/Hostx64/x64',
        'C:/Program Files*/Microsoft Visual Studio/*/Community/VC/Tools/MSVC/*/bin/Hostx64/x64',
        'C:/Program Files*/Microsoft Visual Studio */vc/bin',
+        'C:/ Program Files*/Microsoft Visual Studio/*/Preview/VC/Tools/MSVC/*/bin/Hostx64/x64',
    ]
    for pattern in patterns:
        matches = sorted(glob.glob(pattern))