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
36ebb1d2
Commit
36ebb1d2
authored
Feb 05, 2020
by
rusty1s
Browse files
removed custom atomic op
parent
b957787a
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
0 additions
and
16 deletions
+0
-16
cuda/atomics.cuh
cuda/atomics.cuh
+0
-15
cuda/fps_kernel.cu
cuda/fps_kernel.cu
+0
-1
No files found.
cuda/atomics.cuh
deleted
100644 → 0
View file @
b957787a
#pragma once
#if defined(__CUDA_ARCH__) && (__CUDA_ARCH__ < 600 || CUDA_VERSION < 8000)
static
inline
__device__
void
atomicAdd
(
double
*
address
,
double
val
)
{
unsigned
long
long
int
*
address_as_ull
=
(
unsigned
long
long
int
*
)
address
;
unsigned
long
long
int
old
=
*
address_as_ull
;
unsigned
long
long
int
assumed
;
do
{
assumed
=
old
;
old
=
atomicCAS
(
address_as_ull
,
assumed
,
__double_as_longlong
(
val
+
__longlong_as_double
(
assumed
)));
}
while
(
assumed
!=
old
);
}
#endif
cuda/fps_kernel.cu
View file @
36ebb1d2
#include <ATen/ATen.h>
#include <ATen/ATen.h>
#include "atomics.cuh"
#include "compat.cuh"
#include "compat.cuh"
#include "utils.cuh"
#include "utils.cuh"
...
...
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