Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!49340 code check for ApplyProximalGradientDescentCpuKernel 

Merge pull request !49340 from Yanzhi_YI/codecheck
This commit is contained in:
i-robot 2023-02-24 06:28:02 +00:00 committed by Gitee
parent d3028021ed 56e42fbf2c
commit 4b17adebf8
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/plugin/device/cpu/kernel/apply_proximal_gradient_descent_cpu_kernel.cc
View File

@ -162,7 +162,7 @@ void ApplyProximalGradientDescentCpuKernelMod::LaunchKernelDefault(const std::ve
auto delta_addr = reinterpret_cast<T *>(inputs[4]->addr);
auto task = [this, &var_addr, &alpha_addr, &l1_addr, &l2_addr, &delta_addr](size_t start, size_t end) {
auto cur_input_elements = end - start;
for (int64_t b = 0; b < batch_size_; b++) {
for (size_t b = 0; b < static_cast<size_t>(batch_size_); b++) {
auto offset = b * input_elements_ + start;
auto var_cur = var_addr + offset;
auto delta_cur = delta_addr + offset;
@ -192,7 +192,7 @@ void ApplyProximalGradientDescentCpuKernelMod::LaunchKernelOptFp32(const std::ve
auto task = [this, &var, &alpha, &l1, &l2, &delta](size_t start, size_t end) {
auto cur_input_elements = end - start;
for (int64_t b = 0; b < batch_size_; b++) {
for (size_t b = 0; b < static_cast<size_t>(batch_size_); b++) {
auto offset = b * input_elements_ + start;
auto var_cur = var + offset;
auto delta_cur = delta + offset;