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

!48695 【codex】 

Merge pull request !48695 from KXiong/branch_ger
This commit is contained in:
i-robot 2023-02-13 07:15:49 +00:00 committed by Gitee
parent e1bf554678 419292f91f
commit 16587dc343
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/cpu/kernel/ger_cpu_kernel.cc
View File

@ -250,7 +250,7 @@ bool GerCpuKernelMod::LaunchMacBatches(const std::vector<kernel::AddressPtr> &in
T *input2 = reinterpret_cast<T *>(inputs[kIndex1]->addr); T *input2 = reinterpret_cast<T *>(inputs[kIndex1]->addr);
T *output = reinterpret_cast<T *>(outputs[kIndex0]->addr); T *output = reinterpret_cast<T *>(outputs[kIndex0]->addr);
auto task = [this, &input1, &input2, &output](size_t start, size_t end) { auto task = [this, &input1, &input2, &output](size_t start, size_t end) {
for (size_t batch_index = 0; batch_index < end; batch_index++) { for (size_t batch_index = start; batch_index < end; batch_index++) {
size_t row_i_s = batch_index * this->in1dim_; size_t row_i_s = batch_index * this->in1dim_;
size_t col_i_s = batch_index * this->in2dim_; size_t col_i_s = batch_index * this->in2dim_;
size_t out_i_s = batch_index * this->in1dim_ * this->in2dim_; size_t out_i_s = batch_index * this->in1dim_ * this->in2dim_;