From 1615317f50b112adf23b43a63d2946e7a78030f2 Mon Sep 17 00:00:00 2001 From: jiaorui Date: Thu, 4 Aug 2022 09:42:31 +0800 Subject: [PATCH] clean code overflow exception --- .../ascend/optimizer/enhancer/concat_outputs_for_all_gather.cc | 2 +- .../plugin/device/cpu/kernel/eigen/random_poisson_cpu_kernel.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/concat_outputs_for_all_gather.cc b/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/concat_outputs_for_all_gather.cc index 7edda2b0162..cfc2e2851cb 100644 --- a/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/concat_outputs_for_all_gather.cc +++ b/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/concat_outputs_for_all_gather.cc @@ -205,7 +205,7 @@ AnfNodePtr ConcatOutputsForAllGather::InsertConcatForOutput(const FuncGraphPtr & while (concat_input_num - cur_input_index >= static_cast(inputs_divisor_)) { concat = CreateNewConcat(func_graph, concat_input_nodes, concat_input_info, cur_input_index, inputs_divisor_); output_nodes.push_back(concat); - cur_input_index += inputs_divisor_; + cur_input_index += LongToSize(inputs_divisor_); } size_t rest_num = concat_input_num - cur_input_index; if (rest_num == 1) { diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/random_poisson_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/random_poisson_cpu_kernel.cc index 8c30c57d4d1..6735a012fd9 100644 --- a/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/random_poisson_cpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/random_poisson_cpu_kernel.cc @@ -34,7 +34,7 @@ using KernelRunFunc = RandomPoissonCpuKernelMod::KernelRunFunc; &RandomPoissonCpuKernelMod::LaunchKernel \ } -static unsigned int s_seed = LongToUint(time(nullptr)); +static unsigned int s_seed = static_cast(time(nullptr)); EIGEN_DEVICE_FUNC uint64_t get_random_seed() { auto rnd = rand_r(&s_seed); return IntToSize(rnd);