Custom action space

[alexeybelkov]

❓ Questions and Help

By now CompilerGym has an action space that represents LLVM optimization transforms. I want to add a new action space, based on POSET-RL, i.e. changing an action from being a single transform to a tuple of transforms. But I can't understand how to do this without rewriting the entire LLVMEnv class

[alexeybelkov]

Oh, I just carefully went once more though docs and found commandline_to_actions I guess this the solution

[ChrisCummins]

changing an action from being a single transform to a tuple of transforms

IIUC, the easiest way to do this may be using a wrapper. I.e. wrap the Llvm environment in a class that exposes your new higher-level action space which translates to the underlying space. You may want to take a look at some of the existing wrappers to see if they can be adapted. E.g.

https://github.com/facebookresearch/CompilerGym/blob/development/compiler_gym/wrappers/commandline.py#L105-L108

[alexeybelkov]

changing an action from being a single transform to a tuple of transforms

IIUC, the easiest way to do this may be using a wrapper. I.e. wrap the Llvm environment in a class that exposes your new higher-level action space which translates to the underlying space. You may want to take a look at some of the existing wrappers to see if they can be adapted. E.g.

https://github.com/facebookresearch/CompilerGym/blob/development/compiler_gym/wrappers/commandline.py#L105-L108

Thanks! I'll try this