Browse Source

!682 Fixing UpdateArgTable for ops that compile with L2 enabled

From: @xchu42
Reviewed-by: @ji_chen,@wqtshg
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
337389e4b1
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      ge/single_op/task/op_task.cc

+ 4
- 4
ge/single_op/task/op_task.cc View File

@@ -119,11 +119,11 @@ Status OpTask::DoUpdateArgTable(const SingleOpModelParam &param, bool keep_works
uintptr_t *arg_base = nullptr;
size_t arg_num = 0;
GetIoAddr(arg_base, arg_num);
if (arg_num != all_addresses.size()) {
GELOGE(INTERNAL_ERROR, "[%s] arg number mismatches, expect = %zu, but got = %zu",
if (arg_num < all_addresses.size()) {
GELOGE(INTERNAL_ERROR, "[%s] arg number mismatches, expect at least = %zu, but got = %zu",
op_desc_->GetName().c_str(),
arg_num,
all_addresses.size());
all_addresses.size(),
arg_num);
return INTERNAL_ERROR;
}



Loading…
Cancel
Save