From 2dcdd05698f7aecaac737989bc1e2c044b38554c Mon Sep 17 00:00:00 2001
From: EthenJ <EthenJ@outlook.sg>
Date: Sun, 29 Sep 2024 15:11:43 +0800
Subject: [PATCH] fix building with cuda 12.x (NVIDIA/MinkowskiEngine#601)

---
 src/3rdparty/concurrent_unordered_map.cuh | 1 +
 src/convolution_kernel.cuh                | 2 ++
 src/coordinate_map_gpu.cu                 | 3 +++
 src/spmm.cu                               | 4 ++++
 4 files changed, 10 insertions(+)

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