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
1770a342
"src/include/vscode:/vscode.git/clone" did not exist on "5ff894195c9cc86bac0bb072e4e914ee87a8cbbe"
Commit
1770a342
authored
Jun 13, 2022
by
Paul
Browse files
Format
parent
aeb60bce
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
7 deletions
+7
-7
src/include/migraphx/module.hpp
src/include/migraphx/module.hpp
+1
-4
src/targets/gpu/fuse_mlir.cpp
src/targets/gpu/fuse_mlir.cpp
+6
-3
No files found.
src/include/migraphx/module.hpp
View file @
1770a342
...
...
@@ -178,10 +178,7 @@ struct module
std
::
unique_ptr
<
module_impl
>
impl
;
};
inline
module
&
get_module
(
module
&
m
)
{
return
m
;
}
inline
module
&
get_module
(
module
&
m
)
{
return
m
;
}
}
// namespace MIGRAPHX_INLINE_NS
}
// namespace migraphx
...
...
src/targets/gpu/fuse_mlir.cpp
View file @
1770a342
...
...
@@ -65,9 +65,9 @@ struct find_conv_pointwise
mm
->
set_bypass
();
std
::
unordered_map
<
instruction_ref
,
instruction_ref
>
param_map
;
auto
x
=
mm
->
add_parameter
(
"x"
+
std
::
to_string
(
names
.
size
()),
conv_ins
->
inputs
().
at
(
0
)
->
get_shape
());
conv_ins
->
inputs
().
at
(
0
)
->
get_shape
());
auto
w
=
mm
->
add_parameter
(
"x"
+
std
::
to_string
(
names
.
size
()
+
1
),
conv_ins
->
inputs
().
at
(
1
)
->
get_shape
());
conv_ins
->
inputs
().
at
(
1
)
->
get_shape
());
auto
conv
=
mm
->
add_instruction
(
conv_ins
->
get_operator
(),
{
x
,
w
});
std
::
transform
(
names
.
begin
(),
names
.
end
(),
...
...
@@ -93,7 +93,10 @@ struct find_conv_pointwise
};
}
// namespace
void
fuse_mlir
::
apply
(
module_pass_manager
&
mpm
)
const
{
match
::
find_matches
(
mpm
,
find_conv_pointwise
{});
}
void
fuse_mlir
::
apply
(
module_pass_manager
&
mpm
)
const
{
match
::
find_matches
(
mpm
,
find_conv_pointwise
{});
}
}
// namespace gpu
...
...
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