From b6eb47c55eb677e49029704fdf94af30140f3d3e Mon Sep 17 00:00:00 2001
From: HAOCHENYE <21724054@zju.edu.cn>
Date: Wed, 4 Jan 2023 16:44:22 +0800
Subject: [PATCH] minor refine

---
 mmcv/ops/csrc/common/pytorch_gdb.cpp | 1 +
 setup.py                             | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/mmcv/ops/csrc/common/pytorch_gdb.cpp b/mmcv/ops/csrc/common/pytorch_gdb.cpp
index 53f5004e8..b6021962d 100644
--- a/mmcv/ops/csrc/common/pytorch_gdb.cpp
+++ b/mmcv/ops/csrc/common/pytorch_gdb.cpp
@@ -1,6 +1,7 @@
 #include <torch/csrc/DynamicTypes.h>
 #include <torch/csrc/autograd/variable.h>
 #include <torch/csrc/python_headers.h>
+#include <torch/csrc/autograd/python_variable.h>
 #include <torch/csrc/utils/invalid_arguments.h>
 #include <torch/csrc/utils/python_strings.h>
 #include <torch/csrc/utils/python_tuples.h>
diff --git a/setup.py b/setup.py
index c947a19d0..8e34b8105 100644
--- a/setup.py
+++ b/setup.py
@@ -351,7 +351,8 @@ def get_extensions():
         else:
             print(f'Compiling {ext_name} only with CPU')
             op_files = glob.glob('./mmcv/ops/csrc/pytorch/*.cpp') + \
-                glob.glob('./mmcv/ops/csrc/pytorch/cpu/*.cpp')
+                glob.glob('./mmcv/ops/csrc/pytorch/cpu/*.cpp') + \
+                glob.glob('./mmcv/ops/csrc/common/*.cpp')
             extension = CppExtension
             include_dirs.append(os.path.abspath('./mmcv/ops/csrc/common'))