From d96d59d47b8de3bd0a86afd50f63945d4fb7b3f9 Mon Sep 17 00:00:00 2001 From: machenmusik Date: Tue, 17 Oct 2023 20:54:21 -0400 Subject: [PATCH] Avoid nvcc compilation error (#55) Added define as per https://github.com/facebookresearch/pytorch3d/issues/1614#issuecomment-1720411236 Co-authored-by: Chen, Michael --- setup.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/setup.py b/setup.py index 7eeccee73..dafdd20d7 100644 --- 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",