diff --git a/ge/single_op/single_op.cc b/ge/single_op/single_op.cc index 0daf243e..9b74039e 100755 --- a/ge/single_op/single_op.cc +++ b/ge/single_op/single_op.cc @@ -165,7 +165,7 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY Status SingleOp::ExecuteAsync(c return ret; } - int index = 0; + uint32_t index = 0; for (auto &task : tasks_) { ret = task->LaunchKernel(stream_); if (ret != SUCCESS) { @@ -180,7 +180,7 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY Status SingleOp::ExecuteAsync(c return ret; } -Status SingleOp::ProfilingTaskInfo(int index) { +Status SingleOp::ProfilingTaskInfo(uint32_t index) { if (op_name_.size() <= index) { GELOGE(ACL_ERROR_GE_PARAM_INVALID, "index[%d] is out of range of op_name_ size[%d].", index, op_name_.size()); return ACL_ERROR_GE_PARAM_INVALID; diff --git a/ge/single_op/single_op.h b/ge/single_op/single_op.h index 91db5448..ecc532de 100755 --- a/ge/single_op/single_op.h +++ b/ge/single_op/single_op.h @@ -42,7 +42,7 @@ class SingleOp { Status ValidateArgs(const std::vector &inputs, const std::vector &outputs); Status UpdateArgs(const std::vector &inputs, const std::vector &outputs); Status GetArgs(const std::vector &inputs, const std::vector &outputs); - Status ProfilingTaskInfo(int index); + Status ProfilingTaskInfo(uint32_t index); friend class SingleOpModel; std::mutex *stream_mutex_;