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
composable_kernel
Commits
fd76c787
Commit
fd76c787
authored
Nov 09, 2022
by
Qianfeng Zhang
Browse files
Fix comments again
parent
e6dce0db
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
include/ck/tensor_operation/gpu/device/impl/device_batchnorm_backward_impl.hpp
...ration/gpu/device/impl/device_batchnorm_backward_impl.hpp
+2
-2
No files found.
include/ck/tensor_operation/gpu/device/impl/device_batchnorm_backward_impl.hpp
View file @
fd76c787
...
...
@@ -376,13 +376,13 @@ struct DeviceBatchNormBwdImpl
index_t
space_sz
;
// setup buffer for the partial reduced result for scale
_diff
// setup buffer for the partial reduced result for
d
scale
pArg_
->
workspace_reduce_dscale
=
pArg_
->
p_workspace_
;
space_sz
=
pArg_
->
invariant_length
*
pArg_
->
blkGroupSize
*
sizeof
(
ScaleDataType
);
space_sz
=
math
::
integer_least_multiple
(
space_sz
,
64
);
// setup buffer for the partial reduced result for bias
_diff
// setup buffer for the partial reduced result for
d
bias
pArg_
->
workspace_reduce_dbias
=
reinterpret_cast
<
char
*>
(
pArg_
->
workspace_reduce_dscale
)
+
space_sz
;
...
...
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