From 9f61ba7254eb39d02632aec231e37d5c9277410e Mon Sep 17 00:00:00 2001
From: Chris Cummins <cummins@fb.com>
Date: Tue, 19 Apr 2022 15:50:51 -0700
Subject: [PATCH] Add missing cmake dependencies.

---
 compiler_gym/service/BUILD          |  2 +-
 compiler_gym/service/CMakeLists.txt | 15 +++++++++++++--
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/compiler_gym/service/BUILD b/compiler_gym/service/BUILD
index 92733ef76d..cc127f47a4 100644
--- a/compiler_gym/service/BUILD
+++ b/compiler_gym/service/BUILD
@@ -13,7 +13,7 @@ py_library(
         ":compilation_session",
         ":connection",
         # TODO(github.com/facebookresearch/CompilerGym/pull/633):
-        # add this after circular dependencies are resolved
+        # add this after circular dependencies are resolved:
         # ":client_service_compiler_env",
         ":connection_pool",
         # TODO(github.com/facebookresearch/CompilerGym/pull/633):
diff --git a/compiler_gym/service/CMakeLists.txt b/compiler_gym/service/CMakeLists.txt
index 7971b3513a..557fc3bf4b 100644
--- a/compiler_gym/service/CMakeLists.txt
+++ b/compiler_gym/service/CMakeLists.txt
@@ -14,8 +14,9 @@ cg_py_library(
     ::compilation_session
     ::connection
     # TODO(github.com/facebookresearch/CompilerGym/pull/633):
-    # add this after circular dependencies are resolved
-    #::client_service_compiler_env
+    # add this after circular dependencies are resolved:
+    # ::client_service_compiler_env
+    ::connection_pool
     compiler_gym::errors::errors
     compiler_gym::service::proto::proto
   PUBLIC
@@ -76,3 +77,13 @@ cg_py_library(
     compiler_gym::views::views
   PUBLIC
 )
+
+cg_py_library(
+  NAME
+    connection_pool
+  SRCS
+    "connection_pool.py"
+  DEPS
+    ::connection
+  PUBLIC
+)