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/coordinate_map_gpu.cu b/src/coordinate_map_gpu.cu
index fb7325d1..500cea6b 100644
--- a/src/coordinate_map_gpu.cu
+++ b/src/coordinate_map_gpu.cu
@@ -30,7 +30,8 @@
 #include "kernel_map.cuh"
 #include "kernel_map.hpp"
 #include "sharedmem.cuh"
-
+#include <thrust/remove.h>
+#include <thrust/unique.h>
 #include <thrust/copy.h>
 #include <thrust/execution_policy.h>
 #include <thrust/iterator/counting_iterator.h>
diff --git a/src/local_pooling_transpose_gpu.cu b/src/local_pooling_transpose_gpu.cu
index 5f4dc866..a07d8629 100644
--- a/src/local_pooling_transpose_gpu.cu
+++ b/src/local_pooling_transpose_gpu.cu
@@ -39,6 +39,7 @@
 
 #include <pybind11/pybind11.h>
 #include <torch/extension.h>
+#include <thrust/execution_policy.h>
 
 namespace minkowski {
 
diff --git a/src/spmm.cu b/src/spmm.cu
index 8891a566..8ef224b8 100644
--- a/src/spmm.cu
+++ b/src/spmm.cu
@@ -28,7 +28,9 @@
 #include "math_functions.cuh"
 
 #include <cusparse.h>
-
+#include <thrust/reduce.h>
+#include <thrust/sort.h>
+#include <thrust/execution_policy.h>
 #include <ATen/core/Tensor.h>
 #include <ATen/cuda/CUDAContext.h>
 #include <ATen/cuda/CUDAUtils.h>