Add es gin config files

Author: salaast

compiler_opt/es/gin_configs/blackbox_learner.gin

@@ -0,0 +1,17 @@
+import compiler_opt.es.blackbox_learner
+import compiler_opt.rl.gin_external_configurables
+import compiler_opt.es.blackbox_optimizers
+
+# Blackbox learner config
+BlackboxLearnerConfig.total_steps = 20000
+BlackboxLearnerConfig.blackbox_optimizer = %blackbox_optimizers.Algorithm.MONTE_CARLO
+BlackboxLearnerConfig.est_type = %blackbox_optimizers.EstimatorType.ANTITHETIC
+BlackboxLearnerConfig.fvalues_normalization = True
+BlackboxLearnerConfig.hyperparameters_update_method = %blackbox_optimizers.UpdateMethod.NO_METHOD
+BlackboxLearnerConfig.num_top_directions = 0
+BlackboxLearnerConfig.num_ir_repeats_within_worker = 1
+BlackboxLearnerConfig.num_ir_repeats_across_worker = 1
+BlackboxLearnerConfig.num_exact_evals = 64
+BlackboxLearnerConfig.total_num_perturbations = 384
+BlackboxLearnerConfig.precision_parameter = 0.025
+BlackboxLearnerConfig.step_size = 0.005

compiler_opt/es/gin_configs/blackbox_trainer.gin

@@ -0,0 +1,11 @@
+import compiler_opt.es.es_worker
+# List of flags to add to clang compilation command. The flag names should
+# match the actual flags provided to clang. An example for AFDO reinjection:
+# train_eval.additional_flags=('-fprofile-sample-use=/path/to/gwp.afdo','-fprofile-remapping-file=/path/to/prof_remap.txt')
+train.additional_compilation_flags=()
+
+# List of flags to remove from clang compilation command. The flag names
+# should match the actual flags provided to clang.'
+train.delete_compilation_flags=('-split-dwarf-file','-split-dwarf-output','-fthinlto-index','-fprofile-sample-use','-fprofile-remapping-file')
+
+train.worker_class=@es_worker.ESWorker

compiler_opt/es/gin_configs/inlining.gin

@@ -0,0 +1,16 @@
+import compiler_opt.rl.gin_external_configurables
+import compiler_opt.rl.inlining.config
+
+include 'compiler_opt/rl/inlining/gin_configs/common.gin'
+
+# Inlining model settings
+ActorDistributionNetwork.preprocessing_combiner=@tf.keras.layers.Concatenate()
+ActorDistributionNetwork.fc_layer_params=(64, 64, 64, 64)
+ActorDistributionNetwork.dropout_layer_params=None
+ActorDistributionNetwork.activation_fn=@tf.keras.activations.relu
+
+inlining.config.get_observation_processing_layer_creator.quantile_file_dir='/cns/oz-d/home/mlcompileropt-dev/llvm_inlining/muppet_20210707/vocab/'
+inlining.config.get_observation_processing_layer_creator.with_sqrt = False
+inlining.config.get_observation_processing_layer_creator.with_z_score_normalization = False
+
+policy_utils.create_actor_policy.actor_network_ctor = @actor_distribution_network.ActorDistributionNetwork

compiler_opt/es/gin_configs/inlining_o3.gin

@@ -0,0 +1,16 @@
+import compiler_opt.rl.gin_external_configurables
+import compiler_opt.rl.inlining.config
+
+include 'compiler_opt/rl/inlining/gin_configs/common.gin'
+
+# Inlining model settings
+ActorDistributionNetwork.preprocessing_combiner=@tf.keras.layers.Concatenate()
+ActorDistributionNetwork.fc_layer_params=(64, 64, 64, 64)
+ActorDistributionNetwork.dropout_layer_params=None
+ActorDistributionNetwork.activation_fn=@tf.keras.activations.relu
+
+inlining.config.get_observation_processing_layer_creator.quantile_file_dir='/cns/oz-d/home/mlcompileropt-dev/llvm_inlining/vocab/'
+inlining.config.get_observation_processing_layer_creator.with_sqrt = False
+inlining.config.get_observation_processing_layer_creator.with_z_score_normalization = False
+
+policy_utils.create_actor_policy.actor_network_ctor = @actor_distribution_network.ActorDistributionNetwork

compiler_opt/es/gin_configs/regalloc.gin

@@ -0,0 +1,16 @@
+import compiler_opt.rl.regalloc.config
+import compiler_opt.rl.regalloc.regalloc_network
+
+include 'compiler_opt/rl/regalloc/gin_configs/common.gin'
+
+# Regalloc model settings
+RegAllocNetwork.preprocessing_combiner=@tf.keras.layers.Concatenate()
+RegAllocNetwork.fc_layer_params=(64, 64, 64, 64)
+RegAllocNetwork.dropout_layer_params=None
+[email protected]
+
+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
+
+policy_utils.create_actor_policy.actor_network_ctor = @regalloc_network.RegAllocNetwork