Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-cluster
Commits
b957787a
Commit
b957787a
authored
Feb 05, 2020
by
rusty1s
Browse files
remove deprecation warnings
parent
6b0c170e
Changes
7
Show whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
14 additions
and
7 deletions
+14
-7
cuda/fps.cpp
cuda/fps.cpp
+2
-1
cuda/graclus.cpp
cuda/graclus.cpp
+2
-1
cuda/grid.cpp
cuda/grid.cpp
+2
-1
cuda/knn.cpp
cuda/knn.cpp
+2
-1
cuda/nearest.cpp
cuda/nearest.cpp
+2
-1
cuda/radius.cpp
cuda/radius.cpp
+2
-1
cuda/rw.cpp
cuda/rw.cpp
+2
-1
No files found.
cuda/fps.cpp
View file @
b957787a
#include <torch/extension.h>
#include <torch/extension.h>
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
#define CHECK_CUDA(x) \
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
at
::
Tensor
fps_cuda
(
at
::
Tensor
x
,
at
::
Tensor
batch
,
float
ratio
,
bool
random
);
at
::
Tensor
fps_cuda
(
at
::
Tensor
x
,
at
::
Tensor
batch
,
float
ratio
,
bool
random
);
...
...
cuda/graclus.cpp
View file @
b957787a
#include <torch/extension.h>
#include <torch/extension.h>
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
#define CHECK_CUDA(x) \
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
at
::
Tensor
graclus_cuda
(
at
::
Tensor
row
,
at
::
Tensor
col
,
int64_t
num_nodes
);
at
::
Tensor
graclus_cuda
(
at
::
Tensor
row
,
at
::
Tensor
col
,
int64_t
num_nodes
);
...
...
cuda/grid.cpp
View file @
b957787a
#include <torch/extension.h>
#include <torch/extension.h>
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
#define CHECK_CUDA(x) \
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
at
::
Tensor
grid_cuda
(
at
::
Tensor
pos
,
at
::
Tensor
size
,
at
::
Tensor
start
,
at
::
Tensor
grid_cuda
(
at
::
Tensor
pos
,
at
::
Tensor
size
,
at
::
Tensor
start
,
at
::
Tensor
end
);
at
::
Tensor
end
);
...
...
cuda/knn.cpp
View file @
b957787a
#include <torch/extension.h>
#include <torch/extension.h>
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
#define CHECK_CUDA(x) \
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
at
::
Tensor
knn_cuda
(
at
::
Tensor
x
,
at
::
Tensor
y
,
size_t
k
,
at
::
Tensor
batch_x
,
at
::
Tensor
knn_cuda
(
at
::
Tensor
x
,
at
::
Tensor
y
,
size_t
k
,
at
::
Tensor
batch_x
,
...
...
cuda/nearest.cpp
View file @
b957787a
#include <torch/extension.h>
#include <torch/extension.h>
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
#define CHECK_CUDA(x) \
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
at
::
Tensor
nearest_cuda
(
at
::
Tensor
x
,
at
::
Tensor
y
,
at
::
Tensor
batch_x
,
at
::
Tensor
nearest_cuda
(
at
::
Tensor
x
,
at
::
Tensor
y
,
at
::
Tensor
batch_x
,
...
...
cuda/radius.cpp
View file @
b957787a
#include <torch/extension.h>
#include <torch/extension.h>
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
#define CHECK_CUDA(x) \
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
at
::
Tensor
radius_cuda
(
at
::
Tensor
x
,
at
::
Tensor
y
,
float
radius
,
at
::
Tensor
radius_cuda
(
at
::
Tensor
x
,
at
::
Tensor
y
,
float
radius
,
...
...
cuda/rw.cpp
View file @
b957787a
#include <torch/extension.h>
#include <torch/extension.h>
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
#define CHECK_CUDA(x) \
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
#define IS_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " is not contiguous");
at
::
Tensor
rw_cuda
(
at
::
Tensor
row
,
at
::
Tensor
col
,
at
::
Tensor
start
,
at
::
Tensor
rw_cuda
(
at
::
Tensor
row
,
at
::
Tensor
col
,
at
::
Tensor
start
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment