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
ef3a45c2
Commit
ef3a45c2
authored
Aug 02, 2023
by
Khalique Ahmed
Browse files
formatting
parent
8639a349
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
src/layout_nhwc.cpp
src/layout_nhwc.cpp
+3
-3
test/verify/test_conv_pooling.cpp
test/verify/test_conv_pooling.cpp
+1
-1
No files found.
src/layout_nhwc.cpp
View file @
ef3a45c2
...
@@ -104,7 +104,7 @@ void transform_convolutions(module& m, bool skip_elim_contiguous)
...
@@ -104,7 +104,7 @@ void transform_convolutions(module& m, bool skip_elim_contiguous)
// auto c = conv;
// auto c = conv;
// auto nchw = m.insert_instruction(ins, make_op("layout", {{"permutation", {0, 1, 2, 3}}}),
// auto nchw = m.insert_instruction(ins, make_op("layout", {{"permutation", {0, 1, 2, 3}}}),
// conv); m.debug_print(); if(not skip_elim_contiguous)
// conv); m.debug_print(); if(not skip_elim_contiguous)
auto
c
=
m
.
insert_instruction
(
ins
,
make_op
(
"contiguous"
),
conv
);
auto
c
=
m
.
insert_instruction
(
ins
,
make_op
(
"contiguous"
),
conv
);
m
.
replace_instruction
(
ins
,
c
);
m
.
replace_instruction
(
ins
,
c
);
}
}
}
}
...
@@ -144,12 +144,12 @@ void layout_nhwc::apply(module_pass_manager& mpm) const
...
@@ -144,12 +144,12 @@ void layout_nhwc::apply(module_pass_manager& mpm) const
mpm
.
run_pass
(
dead_code_elimination
{});
mpm
.
run_pass
(
dead_code_elimination
{});
// mpm.get_module().debug_print();
// mpm.get_module().debug_print();
transform_convolutions
(
mpm
.
get_module
(),
this
->
skip_elim_contiguous
);
transform_convolutions
(
mpm
.
get_module
(),
this
->
skip_elim_contiguous
);
mpm
.
run_pass
(
dead_code_elimination
{});
mpm
.
run_pass
(
dead_code_elimination
{});
// std::cout << "after layout" << std::endl;
// std::cout << "after layout" << std::endl;
// mpm.get_module().debug_print();
// mpm.get_module().debug_print();
// if(not this->skip_elim_contiguous)
// if(not this->skip_elim_contiguous)
mpm
.
run_pass
(
eliminate_contiguous
{
"contiguous"
});
mpm
.
run_pass
(
eliminate_contiguous
{
"contiguous"
});
mpm
.
run_pass
(
dead_code_elimination
{});
mpm
.
run_pass
(
dead_code_elimination
{});
mpm
.
run_pass
(
auto_contiguous
{});
mpm
.
run_pass
(
auto_contiguous
{});
mpm
.
run_pass
(
dead_code_elimination
{});
mpm
.
run_pass
(
dead_code_elimination
{});
...
...
test/verify/test_conv_pooling.cpp
View file @
ef3a45c2
...
@@ -39,7 +39,7 @@ struct test_conv_pooling : verify_program<test_conv_pooling>
...
@@ -39,7 +39,7 @@ struct test_conv_pooling : verify_program<test_conv_pooling>
auto
weights
=
auto
weights
=
mm
->
add_parameter
(
"w"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
4
,
3
,
3
,
3
}});
mm
->
add_parameter
(
"w"
,
migraphx
::
shape
{
migraphx
::
shape
::
float_type
,
{
4
,
3
,
3
,
3
}});
auto
conv
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"convolution"
),
input
,
weights
);
auto
conv
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"convolution"
),
input
,
weights
);
auto
relu
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"relu"
),
conv
);
auto
relu
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"relu"
),
conv
);
auto
pooling
=
mm
->
add_instruction
(
auto
pooling
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"pooling"
,
{{
"mode"
,
migraphx
::
op
::
pooling_mode
::
max
}}),
relu
);
migraphx
::
make_op
(
"pooling"
,
{{
"mode"
,
migraphx
::
op
::
pooling_mode
::
max
}}),
relu
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"relu"
),
pooling
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"relu"
),
pooling
);
...
...
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