File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change 1717from compiler_gym .service import ConnectionOpts
1818from compiler_gym .service .client_service_compiler_env import ClientServiceCompilerEnv
1919from compiler_gym .spaces import Reward
20+ from compiler_gym .service .connection_pool import ServiceConnectionPoolBase
2021from compiler_gym .util .decorators import memoized_property
2122from compiler_gym .util .gym_type_hints import ObservationType , OptionalArgumentValue
2223from compiler_gym .views import ObservationSpaceSpec
@@ -79,6 +80,13 @@ def __init__(
7980 # initialization may time out.
8081 Gcc (bin = gcc_bin )
8182
83+ # NOTE(github.com/facebookresearch/CompilerGym/pull/583): The GCC
84+ # environment stalls on the StartSession() RPC call when service
85+ # connection caching is enabled. I believe this has something to do with
86+ # the runtime code generation, but have not been able to diagnose it
87+ # yet. For now, disable service connection caching for GCC environments.
88+ kwargs ["service_pool" ] = ServiceConnectionPoolBase ()
89+
8290 super ().__init__ (
8391 * args ,
8492 ** kwargs ,
You can’t perform that action at this time.
0 commit comments