Add initial gin configs for regalloc_trace in ES

This patch adds a set of gin configs that can be used for regalloc_trace ES
training. This are intended mostly to be starting points so that we have
a set of options that work upstream as an example and they can be modified
within experiments as needed. They need to be modified before use, such as
by setting binary and basic block trace paths.

Reviewers: mtrofin

Reviewed By: mtrofin

Pull Request: #435

Author: boomanaiden154

compiler_opt/es/regalloc_trace/gin_configs/blackbox_learner.gin

@@ -0,0 +1,36 @@
+import compiler_opt.es.blackbox_learner
+import compiler_opt.rl.gin_external_configurables
+import compiler_opt.es.blackbox_optimizers
+import compiler_opt.es.blackbox_evaluator
+import compiler_opt.es.es_trainer_lib
+import compiler_opt.es.regalloc_trace.regalloc_trace_worker
+
+# Blackbox learner config
+BlackboxLearnerConfig.total_steps = 10000
+BlackboxLearnerConfig.total_num_perturbations = 100
+BlackboxLearnerConfig.blackbox_optimizer = %blackbox_optimizers.Algorithm.MONTE_CARLO
+BlackboxLearnerConfig.estimator_type = %blackbox_optimizers.EstimatorType.ANTITHETIC
+BlackboxLearnerConfig.fvalues_normalization = True
+BlackboxLearnerConfig.hyperparameters_update_method = %blackbox_optimizers.UpdateMethod.NO_METHOD
+
+BlackboxLearnerConfig.num_top_directions = 0
+
+BlackboxLearnerConfig.precision_parameter = 0.5
+
+BlackboxLearnerConfig.step_size = 0.005
+
+blackbox_evaluator.TraceBlackboxEvaluator.bb_trace_path = '<bb trace path>'
+blackbox_evaluator.TraceBlackboxEvaluator.function_index_path = '<function index path>'
+
+BlackboxLearnerConfig.evaluator = @blackbox_evaluator.TraceBlackboxEvaluator
+
+compiler_opt.es.es_trainer_lib.train.worker_class = @RegallocTraceWorker
+# Some flags that need to be deleted for successful compilation of XFDO
+# binaries. This set will need to be modified depending upon your compilation
+# setup.
+compiler_opt.es.es_trainer_lib.train.delete_compilation_flags = ('-fprofile-sample-use', '-split-dwarf-file', '-split-dwarf-output', '-fdebug-compilation-dir', '--warning-suppression-mappings')
+
+RegallocTraceWorker.clang_path = '<clang path>'
+RegallocTraceWorker.basic_block_trace_model_path = '<basic block trace model path>'
+RegallocTraceWorker.thread_count = 128
+RegallocTraceWorker.corpus_path = '<corpus path>'

compiler_opt/es/regalloc_trace/gin_configs/regalloc_trace.gin

@@ -0,0 +1,17 @@
+import compiler_opt.rl.gin_external_configurables
+import compiler_opt.rl.regalloc.config
+import compiler_opt.rl.regalloc.regalloc_network
+
+include 'compiler_opt/rl/regalloc/gin_configs/common.gin'
+
+regalloc.config.get_observation_processing_layer_creator.quantile_file_dir='compiler_opt/rl/regalloc/vocab'
+regalloc.config.get_observation_processing_layer_creator.with_sqrt = False
+regalloc.config.get_observation_processing_layer_creator.with_z_score_normalization = False
+
+RegAllocNetwork.preprocessing_combiner=@tf.keras.layers.Concatenate()
+RegAllocNetwork.fc_layer_params=(80, 40)
+RegAllocNetwork.dropout_layer_params=None
+[email protected]
+
+policy_utils.create_actor_policy.actor_network_ctor = @regalloc_network.RegAllocNetwork
+policy_utils.create_actor_policy.greedy = True