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
50cfbcda
Commit
50cfbcda
authored
Jul 08, 2021
by
Shucai Xiao
Browse files
Merge branch 'develop' of github.com:ROCmSoftwarePlatform/AMDMIGraphX into scatter-op
parents
4a24a2dd
f60c3815
Changes
22
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
48 additions
and
0 deletions
+48
-0
test/verify/test_slice_reverse_step.cpp
test/verify/test_slice_reverse_step.cpp
+24
-0
test/verify/test_slice_step_reverse.cpp
test/verify/test_slice_step_reverse.cpp
+24
-0
No files found.
test/verify/test_slice_reverse_step.cpp
0 → 100644
View file @
50cfbcda
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_slice_reverse_step
:
verify_program
<
test_slice_reverse_step
>
{
migraphx
::
program
create_program
()
const
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
migraphx
::
shape
s
{
migraphx
::
shape
::
int32_type
,
{
7
,
5
}};
auto
x
=
mm
->
add_parameter
(
"x"
,
s
);
auto
slice_out
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"slice"
,
{{
"axes"
,
{
0
,
1
}},
{
"starts"
,
{
0
,
2
}},
{
"ends"
,
{
2
,
-
1
}}}),
x
);
auto
step_out
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"reverse"
,
{{
"axes"
,
{
0
,
1
}}}),
slice_out
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"step"
,
{{
"axes"
,
{
0
,
1
}},
{
"steps"
,
{
2
,
2
}}}),
step_out
);
return
p
;
}
};
test/verify/test_slice_step_reverse.cpp
0 → 100644
View file @
50cfbcda
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
struct
test_slice_step_reverse
:
verify_program
<
test_slice_step_reverse
>
{
migraphx
::
program
create_program
()
const
{
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
migraphx
::
shape
s
{
migraphx
::
shape
::
int32_type
,
{
7
,
5
}};
auto
x
=
mm
->
add_parameter
(
"x"
,
s
);
auto
slice_out
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"slice"
,
{{
"axes"
,
{
0
,
1
}},
{
"starts"
,
{
0
,
2
}},
{
"ends"
,
{
2
,
-
1
}}}),
x
);
auto
step_out
=
mm
->
add_instruction
(
migraphx
::
make_op
(
"step"
,
{{
"axes"
,
{
0
,
1
}},
{
"steps"
,
{
2
,
2
}}}),
slice_out
);
mm
->
add_instruction
(
migraphx
::
make_op
(
"reverse"
,
{{
"axes"
,
{
0
}}}),
step_out
);
return
p
;
}
};
Prev
1
2
Next
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