A Reinforcement Learning Environment for Automatic Code Optimization in the MLIR Compiler