Skip to content

Commit b516fa1

Browse files
committed
[gcc] Disable service connection pool for GCC.
Issue facebookresearch#583.
1 parent 0e6e5ef commit b516fa1

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

compiler_gym/envs/gcc/gcc_env.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from compiler_gym.envs.gcc.gcc import Gcc, GccSpec
1717
from compiler_gym.envs.gcc.gcc_rewards import AsmSizeReward, ObjSizeReward
1818
from compiler_gym.service import ConnectionOpts
19+
from compiler_gym.service.connection_pool import ServiceConnectionPoolBase
1920
from compiler_gym.util.decorators import memoized_property
2021
from 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,

0 commit comments

Comments
 (0)