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
ec1ac8c0
Commit
ec1ac8c0
authored
Feb 15, 2019
by
mei-ye
Browse files
fix test/miopen error
parent
7437ebf4
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
2 deletions
+8
-2
src/targets/gpu/include/migraphx/gpu/machine_model.hpp
src/targets/gpu/include/migraphx/gpu/machine_model.hpp
+4
-1
test/gpu/miopen.cpp
test/gpu/miopen.cpp
+4
-1
No files found.
src/targets/gpu/include/migraphx/gpu/machine_model.hpp
View file @
ec1ac8c0
...
...
@@ -29,6 +29,9 @@ struct op_info
weight_map
[
"hip::load_literal"
]
=
std
::
make_pair
(
1
,
1
);
weight_map
[
"hip::allocate"
]
=
std
::
make_pair
(
0
,
1
);
weight_map
[
"@outline"
]
=
std
::
make_pair
(
0
,
1
);
weight_map
[
"slice"
]
=
std
::
make_pair
(
1
,
1
);
weight_map
[
"squeeze"
]
=
std
::
make_pair
(
1
,
1
);
weight_map
[
"unsqueeze"
]
=
std
::
make_pair
(
1
,
1
);
weight_map
[
"gpu::convolution"
]
=
std
::
make_pair
(
4
,
0
);
weight_map
[
"gpu::conv_bias_relu"
]
=
std
::
make_pair
(
4
,
0
);
weight_map
[
"gpu::pooling"
]
=
std
::
make_pair
(
2
,
0
);
...
...
@@ -45,7 +48,7 @@ struct op_info
}
else
{
return
std
::
make_pair
(
1
,
0
);
return
std
::
make_pair
(
1
,
is_context_free
(
op
)
?
1
:
0
);
}
}
std
::
unordered_map
<
std
::
string
,
std
::
pair
<
int
,
int
>>
weight_map
;
...
...
test/gpu/miopen.cpp
View file @
ec1ac8c0
...
...
@@ -130,7 +130,10 @@ migraphx::argument run_gpu(migraphx::program& p)
EXPECT
(
is_shared
(
ctx
,
p
.
get_context
()));
p
.
dry_run
(
m
);
EXPECT
(
is_shared
(
ctx
,
p
.
get_context
()));
return
migraphx
::
gpu
::
from_gpu
(
p
.
eval
(
m
));
auto
eval
=
p
.
eval
(
m
);
p
.
finish
();
auto
ret_val
=
migraphx
::
gpu
::
from_gpu
(
eval
);
return
ret_val
;
}
template
<
class
V
>
...
...
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