Avoid nvcc compilation error (#55)

Added define as per facebookresearch/pytorch3d#1614 (comment) Co-authored-by: Chen, Michael <[email protected]>
nerfstudio-project · Oct 18, 2023 · d96d59d · d96d59d
1 parent a22056a
commit d96d59d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -81,6 +81,8 @@ def get_extensions():
     else:
         nvcc_flags += ["--expt-relaxed-constexpr"]
     extra_compile_args["nvcc"] = nvcc_flags
+    if sys.platform == "win32":
+        extra_compile_args["nvcc"] += ["-DWIN32_LEAN_AND_MEAN"]
 
     extension = CUDAExtension(
         f"gsplat.csrc",