fix ci

7 years ago · 19152541b2
parent ac7cb949d0
commit 19152541b2
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/operators/lookup_table_op.h
+++ b/paddle/fluid/operators/lookup_table_op.h
@ -103,7 +103,8 @@ class LookupTableKernel : public framework::OpKernel<T> {
          memset(output + i * row_width, 0, row_width * sizeof(T));
        } else {
          PADDLE_ENFORCE_GE(ids[i], 0);
-          auto id_index = table_t.index(ids[i]);
+          auto id_index = table_t.Index(ids[i]);
+          PADDLE_ENFORCE_GE(id_index, 0, "the input key should be exists.");
          memcpy(output + i * row_width, table + id_index * row_width,
                 row_width * sizeof(T));
        }