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

!31599 code clean 

Merge pull request !31599 from chenweifeng/code-clean
This commit is contained in:
i-robot 2022-03-21 03:09:55 +00:00 committed by Gitee
parent 9ef9d646ac a005e9d08b
commit 0027472b8c
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -119,8 +119,8 @@ bool PriorityReplayBufferSampleCpuKernel::Launch(const std::vector<AddressPtr> &
for (size_t transition_index = 0; transition_index < samples.size(); transition_index++) {
const std::vector<AddressPtr> &transition = samples[transition_index];
for (size_t item_index = 0; item_index < schema_.size(); item_index++) {
void *offset =
reinterpret_cast<char *>(outputs[item_index + kTransitionIndex]->addr) + schema_[item_index] * transition_index;
void *offset = reinterpret_cast<uint8_t *>(outputs[item_index + kTransitionIndex]->addr) +
schema_[item_index] * transition_index;
MS_EXCEPTION_IF_CHECK_FAIL(memcpy_s(offset, outputs[item_index + kTransitionIndex]->size,
transition[item_index]->addr, transition[item_index]->size) == EOK,
"memcpy_s() failed.");

2
mindspore/ccsrc/plugin/device/gpu/optimizer/neighbor_exchange_v2_fusion.cc
View File

@ -594,7 +594,7 @@ CNodePtr NeighborExchangeV2Fusion::CreateMiddleConcat(const FuncGraphPtr &graph,
if (concat_dim == kWDim) {
concat_input_all.insert(concat_input_all.end(), all_to_all_v_outputs.begin(), all_to_all_v_outputs.begin() + 1);
} else {
int64_t bottom_num = AllToAllRealIds(4, recv_rank_ids);
int64_t bottom_num = AllToAllRealIds(kRankIdFour, recv_rank_ids);
concat_input_all.insert(concat_input_all.end(), all_to_all_v_outputs.begin() + bottom_num,
all_to_all_v_outputs.begin() + bottom_num + 1);
}