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
b60895cd
Commit
b60895cd
authored
Aug 12, 2022
by
turneram
Browse files
Formatting
parent
5a2b89fc
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
5 deletions
+6
-5
src/simplify_algebra.cpp
src/simplify_algebra.cpp
+6
-5
No files found.
src/simplify_algebra.cpp
View file @
b60895cd
...
...
@@ -814,8 +814,8 @@ struct find_conv_dot_horiz_fusion
m
.
move_instructions
(
arg
,
input
);
}
// TODO: Check if axises match
auto
concat
=
m
.
insert_instruction
(
input
,
make_op
(
"concat"
,
{{
"axis"
,
concat_axis
}}),
args
);
auto
concat
=
m
.
insert_instruction
(
input
,
make_op
(
"concat"
,
{{
"axis"
,
concat_axis
}}),
args
);
auto
batch_size
=
input
->
get_shape
().
lens
().
front
();
auto
sequence_length
=
input
->
get_shape
().
lens
().
at
(
1
);
auto
hidden_size
=
input
->
get_shape
().
lens
().
at
(
2
);
...
...
@@ -840,10 +840,11 @@ struct find_conv_dot_horiz_fusion
{{
"dims"
,
{
batch_size
,
sequence_length
,
hidden_size
*
3
}}}),
fused
);
int64_t
len
=
arg
->
get_shape
().
lens
()[
axis
];
m
.
replace_instruction
(
m
.
replace_instruction
(
arg
,
make_op
(
"slice"
,
{{
"axes"
,
{
axis
}},
{
"starts"
,
{
offset
}},
{
"ends"
,
{
offset
+
len
}}}),
make_op
(
"slice"
,
{{
"axes"
,
{
axis
}},
{
"starts"
,
{
offset
}},
{
"ends"
,
{
offset
+
len
}}}),
fused
);
offset
+=
len
;
}
...
...
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