From a7c1016758f71a098d2503575e8452cba0c67506 Mon Sep 17 00:00:00 2001
From: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Date: Wed, 1 Dec 2021 19:10:11 +0800
Subject: [PATCH] [Fix] Fix compiled error on windows (#1543)
* [Fix] Fix compiled error on windows
* fix typo
---
setup.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/setup.py b/setup.py
index ee08f7a43..1284f9f5d 100644
--- a/setup.py
+++ b/setup.py
@@ -224,7 +224,15 @@ def get_extensions():
os.environ.setdefault('MAX_JOBS', str(cpu_use))
define_macros = []
- extra_compile_args = {}
+
+ # Before PyTorch1.8.0, when compiling CUDA code, `cxx` is a
+ # required key passed to PyTorch. Even if there is no flag passed
+ # to cxx, users also need to pass an empty list to PyTorch.
+ # Since PyTorch1.8.0, it has a default value so users do not need
+ # to pass an empty list anymore.
+ # More details at https://github.com/pytorch/pytorch/pull/45956
+ extra_compile_args = {'cxx': []}
+
# Since the PR (https://github.com/open-mmlab/mmcv/pull/1463) uses
# c++14 features, the argument ['std=c++14'] must be added here.
# However, in the windows environment, some standard libraries