fix bug of num > INT_MAX

6 years ago · 241120d94d
parent dac70ad4c5
commit 241120d94d
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/framework/data_feed.cc
+++ b/paddle/fluid/framework/data_feed.cc
@ -546,7 +546,7 @@ bool MultiSlotDataFeed::CheckFile(const char* filename) {
        VLOG(0) << "please check line<" << instance_cout << "> in file<"
                << filename << ">";
        return false;
-      } else if (errno == ERANGE) {
+      } else if (errno == ERANGE || num > INT_MAX) {
        VLOG(0) << "error: the number of ids greater than INT_MAX";
        VLOG(0) << "please check line<" << instance_cout << "> in file<"
                << filename << ">";