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
4fa1fb22
Commit
4fa1fb22
authored
Jul 27, 2019
by
rusty1s
Browse files
fix cuda bug
parent
acd094e7
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
cuda/rw_kernel.cu
cuda/rw_kernel.cu
+3
-3
No files found.
cuda/rw_kernel.cu
View file @
4fa1fb22
...
...
@@ -18,9 +18,9 @@ __global__ void uniform_rw_kernel(
out
[
n
]
=
start
[
n
];
for
(
ptrdiff_t
l
=
1
;
l
<=
walk_length
;
l
++
)
{
auto
i
=
l
*
numel
+
n
;
auto
cur
=
out
[
(
l
-
1
)
*
numel
+
n
];
out
[
i
]
=
col
[
row
[
cur
]
+
int64_t
(
rand
[
i
]
*
deg
[
cur
])];
auto
i
=
(
l
-
1
)
*
numel
+
n
;
auto
cur
=
out
[
i
];
out
[
l
*
numel
+
n
]
=
col
[
row
[
cur
]
+
int64_t
(
rand
[
i
]
*
deg
[
cur
])];
}
}
}
...
...
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