From b7427070a996712cf3e8b4f22d2511086953922c Mon Sep 17 00:00:00 2001
From: chuxing <chuxing@huawei.com>
Date: Mon, 21 Dec 2020 19:59:14 +0800
Subject: [PATCH] Fixing update arg table for ops compiled with L2 enabled

---
 ge/single_op/task/op_task.cc | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/ge/single_op/task/op_task.cc b/ge/single_op/task/op_task.cc
index 4f64251c..cff95324 100755
--- a/ge/single_op/task/op_task.cc
+++ b/ge/single_op/task/op_task.cc
@@ -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;
   }