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
d3401374
Commit
d3401374
authored
Nov 14, 2023
by
Khalique Ahmed
Browse files
update fuse_ops contiguous_pointwise to check gemm input
parent
76871340
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
0 deletions
+9
-0
src/targets/gpu/fuse_ops.cpp
src/targets/gpu/fuse_ops.cpp
+9
-0
No files found.
src/targets/gpu/fuse_ops.cpp
View file @
d3401374
...
@@ -778,6 +778,15 @@ struct find_contiguous_pointwise
...
@@ -778,6 +778,15 @@ struct find_contiguous_pointwise
{
{
auto
ins
=
r
.
result
;
auto
ins
=
r
.
result
;
auto
pw
=
ins
->
inputs
().
front
();
auto
pw
=
ins
->
inputs
().
front
();
for
(
auto
output
:
ins
->
outputs
())
{
if
(
output
->
name
()
!=
"gpu::gemm"
)
continue
;
auto
pw_strides
=
pw
->
get_shape
().
strides
();
// ensure pointwise shape is still valid for gemm call
if
(
std
::
none_of
(
pw_strides
.
end
()
-
2
,
pw_strides
.
end
(),
[
&
](
auto
i
)
{
return
i
==
1
;
}))
return
;
}
auto
alloc
=
ins
->
inputs
().
back
();
auto
alloc
=
ins
->
inputs
().
back
();
auto
args
=
pw
->
inputs
();
auto
args
=
pw
->
inputs
();
args
.
back
()
=
alloc
;
args
.
back
()
=
alloc
;
...
...
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