mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

code clean

This commit is contained in:
wilfChen 2022-03-21 09:34:13 +08:00
parent 143c305286
commit a005e9d08b
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);
}