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
d4613133
"ts/webui/src/components/common/LogPathChild.tsx" did not exist on "f9580cd140c6caa9426fb58e3dff93e81301b0c4"
Commit
d4613133
authored
Jan 26, 2023
by
Paul
Browse files
Fix mean
parent
616cbd16
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/targets/gpu/compile_gen.cpp
src/targets/gpu/compile_gen.cpp
+1
-1
No files found.
src/targets/gpu/compile_gen.cpp
View file @
d4613133
...
@@ -227,7 +227,7 @@ struct reduce_op
...
@@ -227,7 +227,7 @@ struct reduce_op
auto
reduce_elements
=
s
.
elements
()
/
ins
->
get_shape
().
elements
();
auto
reduce_elements
=
s
.
elements
()
/
ins
->
get_shape
().
elements
();
auto
reduce_type
=
s
.
type
();
auto
reduce_type
=
s
.
type
();
r
.
reduction
=
"op::sum{}"
;
r
.
reduction
=
"op::sum{}"
;
std
::
string
mean
=
"op::mean
{
"
+
std
::
to_string
(
reduce_elements
)
+
"}"
;
std
::
string
mean
=
"op::mean
<
"
+
std
::
to_string
(
reduce_elements
)
+
"
>{
}"
;
// Use float accumulator when reduction size is too large for half
// Use float accumulator when reduction size is too large for half
if
(
reduce_type
==
shape
::
half_type
and
reduce_elements
>
16384
)
if
(
reduce_type
==
shape
::
half_type
and
reduce_elements
>
16384
)
r
.
read
=
"compose("
+
mean
+
", op::convert_to<float>{})"
;
r
.
read
=
"compose("
+
mean
+
", op::convert_to<float>{})"
;
...
...
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