File tree 1 file changed +8
-0
lines changed
1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change 16
16
from compiler_gym .envs .gcc .gcc_rewards import AsmSizeReward , ObjSizeReward
17
17
from compiler_gym .service import ConnectionOpts
18
18
from compiler_gym .service .client_service_compiler_env import ClientServiceCompilerEnv
19
+ from compiler_gym .service .connection_pool import ServiceConnectionPoolBase
19
20
from compiler_gym .util .decorators import memoized_property
20
21
from compiler_gym .util .gym_type_hints import ObservationType
21
22
from compiler_gym .util .gym_type_hints import OptionalArgumentValue
@@ -78,6 +79,13 @@ def __init__(
78
79
# initialization may time out.
79
80
Gcc (bin = gcc_bin )
80
81
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
+
81
89
super ().__init__ (
82
90
* args ,
83
91
** kwargs ,
You can’t perform that action at this time.
0 commit comments