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