Skip to content

Commit

Permalink
https://github.com/NVIDIA/MinkowskiEngine/issues/601
Browse files Browse the repository at this point in the history
  • Loading branch information
EthenJ committed Sep 29, 2024
1 parent 02fc608 commit 74793f0
Show file tree
Hide file tree
Showing 4 changed files with 10 additions and 0 deletions.
1 change: 1 addition & 0 deletions src/3rdparty/concurrent_unordered_map.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@

#include <thrust/pair.h>
#include <thrust/count.h>
#include <thrust/execution_policy.h>

#include <functional>
#include <memory>
Expand Down
2 changes: 2 additions & 0 deletions src/convolution_kernel.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -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>
Expand Down
3 changes: 3 additions & 0 deletions src/coordinate_map_gpu.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand Down
4 changes: 4 additions & 0 deletions src/spmm.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 74793f0

Please sign in to comment.