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
c2c5a632
"git@developer.sourcefind.cn:chenpangpang/parler-tts.git" did not exist on "64cfa64e228268fdf0a53bdca4be869737620593"
Commit
c2c5a632
authored
Jul 07, 2022
by
turneram
Browse files
Formatting
parent
cab8156e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
7 deletions
+10
-7
src/simplify_algebra.cpp
src/simplify_algebra.cpp
+10
-7
No files found.
src/simplify_algebra.cpp
View file @
c2c5a632
...
...
@@ -855,19 +855,22 @@ struct find_gelu_erf
{
static
auto
match_mul1
()
{
return
match
::
name
(
"mul"
)(
args
(
match
::
any
().
bind
(
"x"
),
match
::
skip_broadcasts
(
match
::
name
(
"recip"
))));
return
match
::
name
(
"mul"
)(
args
(
match
::
any
().
bind
(
"x"
),
match
::
skip_broadcasts
(
match
::
name
(
"recip"
))));
}
static
auto
match_erf
()
{
return
match
::
name
(
"erf"
)(
match
::
arg
(
0
)(
match_mul1
()));
}
static
auto
match_add2
()
{
return
match
::
name
(
"add"
)(
args
(
match_erf
(),
match
::
skip_broadcasts
(
match
::
has_value
(
1.0
f
))));
return
match
::
name
(
"add"
)(
args
(
match_erf
(),
match
::
skip_broadcasts
(
match
::
has_value
(
1.0
f
))));
}
static
auto
match_add1
()
{
return
match
::
name
(
"add"
)(
args
(
match
::
skip_broadcasts
(
match
::
is_constant
()),
match
::
name
(
"dot"
)));
return
match
::
name
(
"add"
)(
args
(
match
::
skip_broadcasts
(
match
::
is_constant
()),
match
::
name
(
"dot"
)));
}
static
auto
match_mul2
()
{
return
match
::
name
(
"mul"
)(
match
::
args
(
match_add1
(),
match_add2
()));
}
...
...
@@ -880,8 +883,8 @@ struct find_gelu_erf
void
apply
(
module
&
m
,
const
match
::
matcher_result
&
r
)
const
{
auto
ins
=
r
.
result
;
auto
x
=
r
.
instructions
[
"x"
];
auto
ins
=
r
.
result
;
auto
x
=
r
.
instructions
[
"x"
];
auto
lit
=
m
.
add_literal
(
literal
{
shape
{
x
->
get_shape
().
type
()},
{
1.702
f
}});
auto
mul
=
m
.
insert_instruction
(
...
...
@@ -889,12 +892,12 @@ struct find_gelu_erf
mul
=
m
.
insert_instruction
(
ins
,
make_op
(
"mul"
),
x
,
mul
);
auto
sig
=
m
.
insert_instruction
(
ins
,
make_op
(
"neg"
),
mul
);
auto
one
=
m
.
add_literal
(
literal
{
shape
{
x
->
get_shape
().
type
()},
{
1.0
f
}});
one
=
m
.
insert_instruction
(
one
=
m
.
insert_instruction
(
ins
,
make_op
(
"multibroadcast"
,
{{
"out_lens"
,
x
->
get_shape
().
lens
()}}),
one
);
sig
=
m
.
insert_instruction
(
ins
,
make_op
(
"exp"
),
sig
);
sig
=
m
.
insert_instruction
(
ins
,
make_op
(
"add"
),
one
,
sig
);
sig
=
m
.
insert_instruction
(
ins
,
make_op
(
"div"
),
one
,
sig
);
sig
=
m
.
insert_instruction
(
ins
,
make_op
(
"mul"
),
x
,
sig
);
sig
=
m
.
insert_instruction
(
ins
,
make_op
(
"mul"
),
x
,
sig
);
m
.
replace_instruction
(
ins
,
sig
);
}
};
...
...
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