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
dd68643e
Commit
dd68643e
authored
Oct 28, 2022
by
Khalique Ahmed
Browse files
formatting
parent
26574e25
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
52 additions
and
7 deletions
+52
-7
src/rewrite_gemm.cpp
src/rewrite_gemm.cpp
+7
-7
test/verify/gemm_simple.cpp
test/verify/gemm_simple.cpp
+45
-0
No files found.
src/rewrite_gemm.cpp
View file @
dd68643e
...
...
@@ -78,18 +78,18 @@ void rewrite_gemm::apply(module& m) const
auto
in1_transposed
=
m
.
insert_instruction
(
ins
,
make_op
(
"transpose"
,
{{
"permutation"
,
{
3
,
2
,
1
,
0
}}}),
in1
);
auto
conv
=
m
.
insert_instruction
(
ins
,
make_op
(
"convolution"
),
{
in0_transposed
,
in1_transposed
});
auto
conv_transpose
=
m
.
insert_instruction
(
ins
,
make_op
(
"transpose"
,
{{
"permutation"
,
{
0
,
2
,
3
,
1
}}}),
conv
);
auto
conv
=
m
.
insert_instruction
(
ins
,
make_op
(
"convolution"
),
{
in0_transposed
,
in1_transposed
});
auto
conv_transpose
=
m
.
insert_instruction
(
ins
,
make_op
(
"transpose"
,
{{
"permutation"
,
{
0
,
2
,
3
,
1
}}}),
conv
);
auto
out_lens
=
conv_transpose
->
get_shape
().
lens
();
auto
out_lens
=
conv_transpose
->
get_shape
().
lens
();
auto
conv_transpose_out
=
conv_transpose
;
if
(
out_lens
.
size
()
!=
in_size
)
{
out_lens
.
erase
(
out_lens
.
begin
(),
out_lens
.
begin
()
+
(
out_lens
.
size
()
-
in_size
));
conv_transpose_out
=
m
.
insert_instruction
(
ins
,
make_op
(
"reshape"
,
{{
"dims"
,
out_lens
}}),
conv_transpose
);
conv_transpose_out
=
m
.
insert_instruction
(
ins
,
make_op
(
"reshape"
,
{{
"dims"
,
out_lens
}}),
conv_transpose
);
}
m
.
replace_instruction
(
ins
,
conv_transpose_out
);
}
...
...
test/verify/gemm_simple.cpp
0 → 100644
View file @
dd68643e
/*
* The MIT License (MIT)
*
* Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
gemm_simple
:
verify_program
<
gemm_simple
>
{
migraphx
::
program
create_program
()
const
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
migraphx
::
shape
m1_shape
{
migraphx
::
shape
::
float_type
,
{
1
,
3
,
5
}};
migraphx
::
shape
m2_shape
{
migraphx
::
shape
::
float_type
,
{
1
,
5
,
4
}};
auto
l1
=
mm
->
add_parameter
(
"1"
,
m1_shape
);
auto
l2
=
mm
->
add_parameter
(
"2"
,
m2_shape
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"dot"
),
l1
,
l2
);
return
p
;
}
};
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