Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
524baf05
Commit
524baf05
authored
Jun 19, 2019
by
Paul
Browse files
Fix extra data reads
parent
53793762
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
src/targets/gpu/device/include/migraphx/gpu/device/nary.hpp
src/targets/gpu/device/include/migraphx/gpu/device/nary.hpp
+1
-2
No files found.
src/targets/gpu/device/include/migraphx/gpu/device/nary.hpp
View file @
524baf05
...
@@ -282,7 +282,7 @@ void nary_broadcast_vec_impl(
...
@@ -282,7 +282,7 @@ void nary_broadcast_vec_impl(
pack
(
inputs
.
data
()[
i
]...)([
&
](
auto
...
xs
)
__device__
{
pack
(
inputs
.
data
()[
i
]...)([
&
](
auto
...
xs
)
__device__
{
for
(
std
::
size_t
j
=
0
;
j
<
vec_size
;
j
++
)
for
(
std
::
size_t
j
=
0
;
j
<
vec_size
;
j
++
)
{
{
out
put
.
data
()[
i
]
[
j
]
=
f
(
xs
[
j
]...,
b
);
out
[
j
]
=
f
(
xs
[
j
]...,
b
);
}
}
});
});
output
.
data
()[
i
]
=
out
;
output
.
data
()[
i
]
=
out
;
...
@@ -332,7 +332,6 @@ void nary_broadcast_impl(hipStream_t stream, F f, argument result, argument barg
...
@@ -332,7 +332,6 @@ void nary_broadcast_impl(hipStream_t stream, F f, argument result, argument barg
template
<
class
F
,
class
...
Arguments
>
template
<
class
F
,
class
...
Arguments
>
void
nary_standard_vec_impl
(
hipStream_t
stream
,
F
f
,
argument
result
,
Arguments
...
args
)
void
nary_standard_vec_impl
(
hipStream_t
stream
,
F
f
,
argument
result
,
Arguments
...
args
)
{
{
// assert(x.get_shape().elements() == y.get_shape().elements());
const
auto
&
output_shape
=
result
.
get_shape
();
const
auto
&
output_shape
=
result
.
get_shape
();
visit_all
(
result
,
args
...)([
&
](
auto
output
,
auto
...
inputs
)
{
visit_all
(
result
,
args
...)([
&
](
auto
output
,
auto
...
inputs
)
{
using
type
=
device_type
<
std
::
remove_cv_t
<
typename
decltype
(
output
)
::
value_type
>>
;
using
type
=
device_type
<
std
::
remove_cv_t
<
typename
decltype
(
output
)
::
value_type
>>
;
...
...
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