Fixed 352 (#66783)

MufanColin · web-flow · commit c97f6636c932 · 2024-07-31T10:13:35.000+08:00
diff --git a/paddle/fluid/distributed/ps/service/brpc_ps_server.cc b/paddle/fluid/distributed/ps/service/brpc_ps_server.cc
@@ -178,7 +178,11 @@ int32_t BrpcPsServer::ReceiveFromPServer(int msg_type,
   while (ar.Cursor() < ar.Finish()) {
     data.push_back(ar.Get<std::pair<uint64_t, std::string>>());
   }
-  CHECK(ar.Cursor() == ar.Finish());
+  PADDLE_ENFORCE_EQ(ar.Cursor(),
+                    ar.Finish(),
+                    phi::errors::InvalidArgument(
+                        "Expected 'ar.Cursor()' to be equal to 'ar.Finish()', "
+                        "but found they are not equal."));
   this->_shuffled_ins->Write(std::move(data));
   return 0;
 }

Original file line number	Diff line number	Diff line change
`@@ -178,7 +178,11 @@ int32_t BrpcPsServer::ReceiveFromPServer(int msg_type,`
`178`	`178`	`while (ar.Cursor() < ar.Finish()) {`
`179`	`179`	`data.push_back(ar.Get<std::pair<uint64_t, std::string>>());`
`180`	`180`	`}`
`181`		`- CHECK(ar.Cursor() == ar.Finish());`
	`181`	`+ PADDLE_ENFORCE_EQ(ar.Cursor(),`
	`182`	`+ ar.Finish(),`
	`183`	`+ phi::errors::InvalidArgument(`
	`184`	`+ "Expected 'ar.Cursor()' to be equal to 'ar.Finish()', "`
	`185`	`+ "but found they are not equal."));`
`182`	`186`	`this->_shuffled_ins->Write(std::move(data));`
`183`	`187`	`return 0;`
`184`	`188`	`}`