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
gaoqiong
MIGraphX
Commits
868230f5
"vscode:/vscode.git/clone" did not exist on "52decd2a4346f783a6229603716a08c2640aebe9"
Commit
868230f5
authored
Aug 09, 2022
by
Paul
Browse files
Format
parent
14ec0ee9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
src/targets/gpu/compile_gen.cpp
src/targets/gpu/compile_gen.cpp
+3
-1
No files found.
src/targets/gpu/compile_gen.cpp
View file @
868230f5
...
...
@@ -85,7 +85,9 @@ preload preload::broadcasts(std::size_t axis, const std::vector<shape>& inputs)
std
::
vector
<
bool
>
result
(
inputs
.
size
());
std
::
vector
<
std
::
size_t
>
preloaded
;
auto
idxs
=
range
(
inputs
.
size
());
std
::
copy_if
(
idxs
.
begin
(),
idxs
.
end
(),
std
::
back_inserter
(
preloaded
),
[
&
](
auto
i
){
return
inputs
[
i
].
strides
()[
axis
]
==
0
;
});
std
::
copy_if
(
idxs
.
begin
(),
idxs
.
end
(),
std
::
back_inserter
(
preloaded
),
[
&
](
auto
i
)
{
return
inputs
[
i
].
strides
()[
axis
]
==
0
;
});
std
::
sort
(
preloaded
.
begin
(),
preloaded
.
end
(),
by
(
std
::
less
<>
{},
[
&
](
auto
i
)
{
return
inputs
[
i
].
bytes
();
}));
...
...
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