Fix windows specific crashes

Stability-AI · Aug 29, 2024 · 47a99bd · 47a99bd
1 parent 8a667f1
commit 47a99bd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/texture_baker/setup.py b/texture_baker/setup.py
@@ -1,5 +1,6 @@
 import glob
 import os
+import platform
 
 import torch
 from setuptools import find_packages, setup
@@ -39,6 +40,10 @@ def get_extensions():
     }
     if debug_mode:
         extra_compile_args["cxx"].append("-g")
+        if platform.system() == "Windows":
+            extra_compile_args["cxx"].append("/Z7")
+            extra_compile_args["cxx"].append("/Od")
+            extra_link_args.extend(["/DEBUG"])
         extra_compile_args["cxx"].append("-UNDEBUG")
         extra_compile_args["nvcc"].append("-UNDEBUG")
         extra_compile_args["nvcc"].append("-g")

diff --git a/texture_baker/texture_baker/csrc/baker.cpp b/texture_baker/texture_baker/csrc/baker.cpp
@@ -247,7 +247,8 @@ namespace texture_baker_cpp
         std::iota(triangle_indices.begin(), triangle_indices.end(), 0);
 
         // Build BVH nodes
-        nodes.reserve(triangles.size());
+        // Reserve extra capacity to fix windows specific crashes
+        nodes.reserve(triangles.size()*2+1);
         nodes.push_back({}); // Create the root node
         root = 0;