fix bug of num > INT_MAX

241120d9 · xjqbest · dac70ad4 · 241120d9
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/framework/data_feed.cc paddle/fluid/framework/data_feed.cc +1 -1

未找到文件。
--- 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 << ">";