Collecting failure status progress

1 files changed, 6 insertions, 0 deletions

diff --git a/test/cpp/qps/server_async.cc b/test/cpp/qps/server_async.cc
index 082b4bc72f..1163b765e3 100644
--- a/test/cpp/qps/server_async.cc
+++ b/test/cpp/qps/server_async.cc
@@ -37,6 +37,7 @@
 #include <mutex>
 #include <thread>
 
+#include <grpc++/buffer_pool.h>
 #include <grpc++/generic/async_generic_service.h>
 #include <grpc++/security/server_credentials.h>
 #include <grpc++/server.h>
@@ -95,6 +96,11 @@ class AsyncQpsServerTest : public Server {
       srv_cqs_.emplace_back(builder.AddCompletionQueue());
     }
 
+    if (config.buffer_pool_size() > 0) {
+      builder.SetBufferPool(
+          BufferPool("AsyncQpsServerTest").Resize(config.buffer_pool_size()));
+    }
+
     server_ = builder.BuildAndStart();
 
     using namespace std::placeholders;