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
658c55ff
Commit
658c55ff
authored
Nov 09, 2022
by
Qianfeng Zhang
Browse files
Rename in the comments
parent
f75931c1
Changes
1
Show 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 @
658c55ff
...
...
@@ -335,11 +335,11 @@ struct DeviceBatchNormBwdImpl
if
(
UseMultiblockInK
&&
pArg_
->
blkGroupSize
>
1
)
{
// workspace for the partial reduced result for scale
_diff
// workspace for the partial reduced result for
d
scale
workspace_size
+=
pArg_
->
invariant_length
*
pArg_
->
blkGroupSize
*
sizeof
(
ScaleDataType
)
+
64
;
// workspace for the partial reduced result for bias
_diff
// workspace for the partial reduced result for
d
bias
workspace_size
+=
pArg_
->
invariant_length
*
pArg_
->
blkGroupSize
*
sizeof
(
BiasDataType
)
+
64
;
...
...
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