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-sparse
Commits
99117398
Commit
99117398
authored
Nov 08, 2020
by
rusty1s
Browse files
replace auto
parent
e0c91397
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
csrc/cpu/convert_cpu.cpp
csrc/cpu/convert_cpu.cpp
+6
-6
No files found.
csrc/cpu/convert_cpu.cpp
View file @
99117398
...
@@ -9,17 +9,17 @@ torch::Tensor ind2ptr_cpu(torch::Tensor ind, int64_t M) {
...
@@ -9,17 +9,17 @@ torch::Tensor ind2ptr_cpu(torch::Tensor ind, int64_t M) {
auto
out_data
=
out
.
data_ptr
<
int64_t
>
();
auto
out_data
=
out
.
data_ptr
<
int64_t
>
();
int64_t
numel
=
ind
.
numel
(),
idx
=
ind_data
[
0
],
next_idx
;
int64_t
numel
=
ind
.
numel
(),
idx
=
ind_data
[
0
],
next_idx
;
for
(
auto
i
=
0
;
i
<=
idx
;
i
++
)
for
(
int64_t
i
=
0
;
i
<=
idx
;
i
++
)
out_data
[
i
]
=
0
;
out_data
[
i
]
=
0
;
for
(
auto
i
=
0
;
i
<
numel
-
1
;
i
++
)
{
for
(
int64_t
i
=
0
;
i
<
numel
-
1
;
i
++
)
{
next_idx
=
ind_data
[
i
+
1
];
next_idx
=
ind_data
[
i
+
1
];
for
(
auto
j
=
idx
;
j
<
next_idx
;
j
++
)
for
(
int64_t
j
=
idx
;
j
<
next_idx
;
j
++
)
out_data
[
j
+
1
]
=
i
+
1
;
out_data
[
j
+
1
]
=
i
+
1
;
idx
=
next_idx
;
idx
=
next_idx
;
}
}
for
(
auto
i
=
idx
+
1
;
i
<
M
+
1
;
i
++
)
for
(
int64_t
i
=
idx
+
1
;
i
<
M
+
1
;
i
++
)
out_data
[
i
]
=
numel
;
out_data
[
i
]
=
numel
;
return
out
;
return
out
;
...
@@ -32,9 +32,9 @@ torch::Tensor ptr2ind_cpu(torch::Tensor ptr, int64_t E) {
...
@@ -32,9 +32,9 @@ torch::Tensor ptr2ind_cpu(torch::Tensor ptr, int64_t E) {
auto
out_data
=
out
.
data_ptr
<
int64_t
>
();
auto
out_data
=
out
.
data_ptr
<
int64_t
>
();
int64_t
idx
=
ptr_data
[
0
],
next_idx
;
int64_t
idx
=
ptr_data
[
0
],
next_idx
;
for
(
auto
i
=
0
;
i
<
ptr
.
numel
()
-
1
;
i
++
)
{
for
(
int64_t
i
=
0
;
i
<
ptr
.
numel
()
-
1
;
i
++
)
{
next_idx
=
ptr_data
[
i
+
1
];
next_idx
=
ptr_data
[
i
+
1
];
for
(
auto
e
=
idx
;
e
<
next_idx
;
e
++
)
for
(
int64_t
e
=
idx
;
e
<
next_idx
;
e
++
)
out_data
[
e
]
=
i
;
out_data
[
e
]
=
i
;
idx
=
next_idx
;
idx
=
next_idx
;
}
}
...
...
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