diff --git a/src/3rdparty/concurrent_unordered_map.cuh b/src/3rdparty/concurrent_unordered_map.cuh
index ed8e1b23..b4c0504d 100644
--- a/src/3rdparty/concurrent_unordered_map.cuh
+++ b/src/3rdparty/concurrent_unordered_map.cuh
@@ -27,6 +27,7 @@
 
 #include <thrust/pair.h>
 #include <thrust/count.h>
+#include <thrust/execution_policy.h>
 
 #include <functional>
 #include <memory>
diff --git a/src/convolution_kernel.cuh b/src/convolution_kernel.cuh
index aed72279..9801340e 100644
--- a/src/convolution_kernel.cuh
+++ b/src/convolution_kernel.cuh
@@ -33,6 +33,8 @@
 #include "math_functions.cuh"
 #include "types.hpp"
 
+#include <thrust/execution_policy.h>
+
 namespace minkowski {
 
 template <typename Dtype, typename Itype, typename ByteAllocator>
diff --git a/src/coordinate_map_gpu.cu b/src/coordinate_map_gpu.cu
index fb7325d1..0609b540 100644
--- a/src/coordinate_map_gpu.cu
+++ b/src/coordinate_map_gpu.cu
@@ -37,6 +37,9 @@
 #include <thrust/iterator/transform_iterator.h>
 #include <thrust/sort.h>
 
+#include <thrust/unique.h>
+#include <thrust/remove.h>
+
 namespace minkowski {
 
 namespace detail {
diff --git a/src/spmm.cu b/src/spmm.cu
index 8891a566..4b6d1468 100644
--- a/src/spmm.cu
+++ b/src/spmm.cu
@@ -36,6 +36,10 @@
 #include <torch/extension.h>
 #include <torch/script.h>
 
+#include <thrust/execution_policy.h>
+#include <thrust/reduce.h>
+#include <thrust/sort.h>
+
 namespace minkowski {
 
 #define BLOCK_SIZE 128