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
af8cb16f
"...gpu/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "03f5c679e6d850fc97c3f64a64784e16d226f1f6"
Commit
af8cb16f
authored
Jun 20, 2019
by
Paul
Browse files
Formatting
parent
c2db1916
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
3 deletions
+4
-3
src/targets/gpu/device/include/migraphx/gpu/device/tensor.hpp
...targets/gpu/device/include/migraphx/gpu/device/tensor.hpp
+4
-2
src/targets/gpu/device/include/migraphx/gpu/device/types.hpp
src/targets/gpu/device/include/migraphx/gpu/device/types.hpp
+0
-1
No files found.
src/targets/gpu/device/include/migraphx/gpu/device/tensor.hpp
View file @
af8cb16f
...
...
@@ -263,7 +263,8 @@ auto hip_visit_all(T&& x, Ts&&... xs)
{
return
[
&
](
auto
f
)
{
visit_tensor_size
(
x
.
get_shape
().
lens
().
size
(),
[
&
](
auto
dim
)
{
visit_all
(
x
,
xs
...)([
&
](
auto
...
vs
)
{
f
(
make_hip_tensor_view
<
dim
>
(
device_cast
(
vs
))...);
});
visit_all
(
x
,
xs
...)([
&
](
auto
...
vs
)
{
f
(
make_hip_tensor_view
<
dim
>
(
device_cast
(
vs
))...);
});
});
};
}
...
...
@@ -273,7 +274,8 @@ auto hip_vec_visit_all(T&& x, Ts&&... xs)
{
return
[
&
](
auto
f
)
{
visit_tensor_size
(
x
.
get_shape
().
lens
().
size
(),
[
&
](
auto
dim
)
{
visit_all
(
x
,
xs
...)([
&
](
auto
...
vs
)
{
f
(
make_hip_tensor_view
<
dim
>
(
as_vec
<
N
>
(
device_cast
(
vs
)))...);
});
visit_all
(
x
,
xs
...)(
[
&
](
auto
...
vs
)
{
f
(
make_hip_tensor_view
<
dim
>
(
as_vec
<
N
>
(
device_cast
(
vs
)))...);
});
});
};
}
...
...
src/targets/gpu/device/include/migraphx/gpu/device/types.hpp
View file @
af8cb16f
...
...
@@ -17,7 +17,6 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace
gpu
{
namespace
device
{
template
<
class
T
,
std
::
size_t
N
>
using
vec
=
T
__attribute__
((
ext_vector_type
(
N
)));
...
...
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