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
974098b1
Commit
974098b1
authored
Nov 07, 2018
by
Paul
Browse files
Merge branch 'master' into test-driver
parents
17e573c0
5b888363
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
0 deletions
+16
-0
test/gpu/miopen.cpp
test/gpu/miopen.cpp
+16
-0
No files found.
test/gpu/miopen.cpp
View file @
974098b1
...
@@ -216,6 +216,21 @@ struct test_scale
...
@@ -216,6 +216,21 @@ struct test_scale
}
}
};
};
struct
test_slice
{
migraph
::
program
create_program
()
const
{
migraph
::
program
p
;
migraph
::
shape
s
{
migraph
::
shape
::
int32_type
,
{
2
,
2
,
4
}};
auto
x
=
p
.
add_parameter
(
"x"
,
s
);
auto
y
=
p
.
add_parameter
(
"y"
,
{
migraph
::
shape
::
int32_type
,
{
2
,
2
,
2
}});
auto
slice0
=
p
.
add_instruction
(
migraph
::
op
::
slice
{{
2
},
{
0
},
{
2
}},
x
);
p
.
add_instruction
(
migraph
::
op
::
add
{},
y
,
slice0
);
return
p
;
}
};
struct
test_triadd
struct
test_triadd
{
{
migraph
::
program
create_program
()
const
migraph
::
program
create_program
()
const
...
@@ -840,4 +855,5 @@ int main()
...
@@ -840,4 +855,5 @@ int main()
verify_program
<
test_conv_bn
>
();
verify_program
<
test_conv_bn
>
();
verify_program
<
test_conv_bn_relu_pooling
>
();
verify_program
<
test_conv_bn_relu_pooling
>
();
verify_program
<
test_conv_bn_relu_pooling2
>
();
verify_program
<
test_conv_bn_relu_pooling2
>
();
verify_program
<
test_slice
>
();
}
}
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