Fixing update arg table for ops compiled with L2 enabled

pull/682/head
chuxing 4 years ago
parent bc21fbdb2e
commit b7427070a9

@ -118,11 +118,11 @@ Status OpTask::UpdateArgTable(const SingleOpModelParam &param) {
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