Skip to content

Commit 057d412

Browse files
author
Yifan Yang
committed
[fix & misc] 1. Fix build errors with CUDA 12.x. (refer to: NVIDIA#567). 2. Update README.md for direct installation.
1 parent 02fc608 commit 057d412

File tree

5 files changed

+13
-0
lines changed

5 files changed

+13
-0
lines changed

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,12 @@ If you cannot find a relevant problem, please report the issue on [the github is
7979

8080
### Pip
8181

82+
**Update**: To install `MinkowskiEngine` directly in `./site-packages`, use:
83+
84+
```
85+
python setup.py install
86+
```
87+
8288
The MinkowskiEngine is distributed via [PyPI MinkowskiEngine][pypi-url] which can be installed simply with `pip`.
8389
First, install pytorch following the [instruction](https://pytorch.org). Next, install `openblas`.
8490

src/3rdparty/concurrent_unordered_map.cuh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
#include <iterator>
3636
#include <limits>
3737
#include <type_traits>
38+
#include <thrust/execution_policy.h>
3839

3940
namespace {
4041
template <std::size_t N>

src/convolution_kernel.cuh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
#include "kernel_map.cuh"
3333
#include "math_functions.cuh"
3434
#include "types.hpp"
35+
#include <thrust/execution_policy.h>
3536

3637
namespace minkowski {
3738

src/coordinate_map_gpu.cu

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@
3636
#include <thrust/iterator/counting_iterator.h>
3737
#include <thrust/iterator/transform_iterator.h>
3838
#include <thrust/sort.h>
39+
#include <thrust/unique.h>
40+
#include <thrust/remove.h>
3941

4042
namespace minkowski {
4143

src/spmm.cu

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,9 @@
3535
#include <c10/cuda/CUDACachingAllocator.h>
3636
#include <torch/extension.h>
3737
#include <torch/script.h>
38+
#include <thrust/execution_policy.h>
39+
#include <thrust/reduce.h>
40+
#include <thrust/sort.h>
3841

3942
namespace minkowski {
4043

0 commit comments

Comments
 (0)