From ce930eeb403a8e3f99693662ec5ce329a0ab3528 Mon Sep 17 00:00:00 2001
From: Zhirui Dai <daizhirui@hotmail.com>
Date: Sun, 22 Oct 2023 18:02:01 -0700
Subject: [PATCH] fix build with CUDA 12.2

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

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>