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
10fe3a73
Commit
10fe3a73
authored
Feb 04, 2019
by
Shucai Xiao
Browse files
merge from branch rnn_operator
parents
40babfd5
0cc5b80e
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
12 additions
and
4 deletions
+12
-4
src/targets/cpu/target.cpp
src/targets/cpu/target.cpp
+9
-1
src/targets/gpu/target.cpp
src/targets/gpu/target.cpp
+2
-2
test/gpu/miopen.cpp
test/gpu/miopen.cpp
+1
-1
No files found.
src/targets/cpu/target.cpp
View file @
10fe3a73
...
...
@@ -3,7 +3,11 @@
#include <migraphx/cpu/lowering.hpp>
#include <migraphx/auto_contiguous.hpp>
#include <migraphx/rewrite_rnn.hpp>
<<<<<<<
HEAD
#include <migraphx/rewrite_gru.hpp>
=======
#include <migraphx/dead_code_elimination.hpp>
>>>>>>>
rnn_operator
namespace
migraphx
{
inline
namespace
MIGRAPHX_INLINE_NS
{
...
...
@@ -13,7 +17,11 @@ std::string target::name() const { return "cpu"; }
std
::
vector
<
pass
>
target
::
get_passes
(
migraphx
::
context
&
)
const
{
return
{
auto_contiguous
{},
rewrite_rnn
{},
rewrite_gru
{},
lowering
{}};
return
{
auto_contiguous
{},
rewrite_rnn
{},
dead_code_elimination
{},
lowering
{},
dead_code_elimination
{}};
}
}
// namespace cpu
...
...
src/targets/gpu/target.cpp
View file @
10fe3a73
...
...
@@ -36,8 +36,8 @@ std::vector<pass> target::get_passes(migraphx::context& gctx) const
rewrite_rnn
{},
rewrite_gru
{},
dead_code_elimination
{},
common_subexpression_elimination
{},
dead_code_elimination
{},
//
common_subexpression_elimination{},
//
dead_code_elimination{},
simplify_algebra
{},
dead_code_elimination
{},
constant_propagate
{},
...
...
test/gpu/miopen.cpp
View file @
10fe3a73
...
...
@@ -1542,5 +1542,5 @@ int main()
verify_program
<
test_rnn_5args
>
();
verify_program
<
test_rnn_bidirectional
>
();
verify_program
<
test_rnn_bidirectional10
>
();
//
verify_program<test_rnn_bi_3args>();
verify_program
<
test_rnn_bi_3args
>
();
}
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