Commit fd76c787 authored by Qianfeng Zhang's avatar Qianfeng Zhang
Browse files

Fix comments again

parent e6dce0db
...@@ -376,13 +376,13 @@ struct DeviceBatchNormBwdImpl ...@@ -376,13 +376,13 @@ struct DeviceBatchNormBwdImpl
index_t space_sz; index_t space_sz;
// setup buffer for the partial reduced result for scale_diff // setup buffer for the partial reduced result for dscale
pArg_->workspace_reduce_dscale = pArg_->p_workspace_; pArg_->workspace_reduce_dscale = pArg_->p_workspace_;
space_sz = pArg_->invariant_length * pArg_->blkGroupSize * sizeof(ScaleDataType); space_sz = pArg_->invariant_length * pArg_->blkGroupSize * sizeof(ScaleDataType);
space_sz = math::integer_least_multiple(space_sz, 64); 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 dbias
pArg_->workspace_reduce_dbias = pArg_->workspace_reduce_dbias =
reinterpret_cast<char*>(pArg_->workspace_reduce_dscale) + space_sz; reinterpret_cast<char*>(pArg_->workspace_reduce_dscale) + space_sz;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment