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'))