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 1616from compiler_gym .envs .gcc .gcc_rewards import AsmSizeReward , ObjSizeReward
1717from compiler_gym .service import ConnectionOpts
1818from compiler_gym .service .client_service_compiler_env import ClientServiceCompilerEnv
19+ from compiler_gym .service .connection_pool import ServiceConnectionPoolBase
1920from compiler_gym .util .decorators import memoized_property
2021from compiler_gym .util .gym_type_hints import ObservationType
2122from compiler_gym .util .gym_type_hints import OptionalArgumentValue
@@ -78,6 +79,13 @@ def __init__(
7879 # initialization may time out.
7980 Gcc (bin = gcc_bin )
8081
82+ # NOTE(github.com/facebookresearch/CompilerGym/pull/583): The GCC
83+ # environment stalls on the StartSession() RPC call when service
84+ # connection caching is enabled. I believe this has something to do with
85+ # the runtime code generation, but have not been able to diagnose it
86+ # yet. For now, disable service connection caching for GCC environments.
87+ kwargs ["service_pool" ] = ServiceConnectionPoolBase ()
88+
8189 super ().__init__ (
8290 * args ,
8391 ** kwargs ,
You can’t perform that action at this time.
0 commit comments