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
fb573172
Commit
fb573172
authored
Aug 25, 2022
by
turneram
Browse files
Formatting
parent
8d378877
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
5 deletions
+5
-5
src/include/migraphx/op/ck_elementwise.hpp
src/include/migraphx/op/ck_elementwise.hpp
+2
-3
src/targets/gpu/kernels/include/migraphx/kernels/ck_elementwise.hpp
...s/gpu/kernels/include/migraphx/kernels/ck_elementwise.hpp
+2
-1
test/verify/0ck_elementwise_test.cpp
test/verify/0ck_elementwise_test.cpp
+1
-1
No files found.
src/include/migraphx/op/ck_elementwise.hpp
View file @
fb573172
...
...
@@ -63,9 +63,8 @@ struct ck_elementwise
{
argument
result
{
output_shape
};
visit_all
(
result
,
args
[
0
],
args
[
1
])([
&
](
auto
output
,
auto
input1
,
auto
input2
)
{
par_for
(
output_shape
.
elements
(),
[
&
](
const
auto
i
)
{
output
[
i
]
=
input1
[
i
]
+
input2
[
i
];
});
par_for
(
output_shape
.
elements
(),
[
&
](
const
auto
i
)
{
output
[
i
]
=
input1
[
i
]
+
input2
[
i
];
});
});
return
result
;
}
...
...
src/targets/gpu/kernels/include/migraphx/kernels/ck_elementwise.hpp
View file @
fb573172
...
...
@@ -30,7 +30,8 @@
namespace
migraphx
{
template
<
class
T
,
class
U
,
class
V
>
__device__
void
ck_elementwise
(
const
T
&
/* data_t */
,
const
U
&
/* indices_t */
,
const
V
&
/* output_t */
)
__device__
void
ck_elementwise
(
const
T
&
/* data_t */
,
const
U
&
/* indices_t */
,
const
V
&
/* output_t */
)
{
}
...
...
test/verify/0ck_elementwise_test.cpp
View file @
fb573172
...
...
@@ -34,7 +34,7 @@ struct ck_elementwise : verify_program<ck_elementwise>
migraphx
::
program
p
;
auto
*
mm
=
p
.
get_main_module
();
migraphx
::
shape
m1_shape
{
migraphx
::
shape
::
float_type
,
{
10
,
20
}};
//migraphx::shape m2_shape{migraphx::shape::float_type, {20, 10}};
//
migraphx::shape m2_shape{migraphx::shape::float_type, {20, 10}};
auto
l1
=
mm
->
add_parameter
(
"1"
,
m1_shape
);
auto
l2
=
mm
->
add_parameter
(
"2"
,
m1_shape
);
...
...
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