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
bc476876
"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "2478cb6f840206eb0d78cd27320d9068be483860"
Commit
bc476876
authored
Mar 14, 2020
by
rusty1s
Browse files
typo
parent
9436b36c
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
csrc/cpu/fps_cpu.cpp
csrc/cpu/fps_cpu.cpp
+1
-1
csrc/cuda/fps_cuda.cu
csrc/cuda/fps_cuda.cu
+1
-1
No files found.
csrc/cpu/fps_cpu.cpp
View file @
bc476876
...
@@ -12,7 +12,7 @@ torch::Tensor fps_cpu(torch::Tensor src, torch::Tensor ptr, double ratio,
...
@@ -12,7 +12,7 @@ torch::Tensor fps_cpu(torch::Tensor src, torch::Tensor ptr, double ratio,
CHECK_CPU
(
src
);
CHECK_CPU
(
src
);
CHECK_CPU
(
ptr
);
CHECK_CPU
(
ptr
);
CHECK_INPUT
(
ptr
.
dim
()
==
1
);
CHECK_INPUT
(
ptr
.
dim
()
==
1
);
AT_ASSERTM
(
ratio
>
0
and
ratio
<
1
,
"Invalid input"
);
AT_ASSERTM
(
ratio
>
0
&&
ratio
<
1
,
"Invalid input"
);
src
=
src
.
view
({
src
.
size
(
0
),
-
1
}).
contiguous
();
src
=
src
.
view
({
src
.
size
(
0
),
-
1
}).
contiguous
();
ptr
=
ptr
.
contiguous
();
ptr
=
ptr
.
contiguous
();
...
...
csrc/cuda/fps_cuda.cu
View file @
bc476876
...
@@ -69,7 +69,7 @@ torch::Tensor fps_cuda(torch::Tensor src, torch::Tensor ptr, double ratio,
...
@@ -69,7 +69,7 @@ torch::Tensor fps_cuda(torch::Tensor src, torch::Tensor ptr, double ratio,
CHECK_CUDA
(
src
);
CHECK_CUDA
(
src
);
CHECK_CUDA
(
ptr
);
CHECK_CUDA
(
ptr
);
CHECK_INPUT
(
ptr
.
dim
()
==
1
);
CHECK_INPUT
(
ptr
.
dim
()
==
1
);
AT_ASSERTM
(
ratio
>
0
and
ratio
<
1
,
"Invalid input"
);
AT_ASSERTM
(
ratio
>
0
&&
ratio
<
1
,
"Invalid input"
);
cudaSetDevice
(
src
.
get_device
());
cudaSetDevice
(
src
.
get_device
());
src
=
src
.
view
({
src
.
size
(
0
),
-
1
}).
contiguous
();
src
=
src
.
view
({
src
.
size
(
0
),
-
1
}).
contiguous
();
...
...
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