├── .github └── workflows │ └── main.yml ├── .gitignore ├── .pylintrc ├── Dockerfile ├── LICENSE ├── Pipfile ├── Pipfile.lock ├── README.md ├── check-license.sh ├── compiler_opt ├── __init__.py ├── distributed │ ├── __init__.py │ ├── buffered_scheduler.py │ ├── buffered_scheduler_test.py │ ├── local │ │ ├── __init__.py │ │ ├── local_worker_manager.py │ │ └── local_worker_manager_test.py │ ├── worker.py │ ├── worker_manager.py │ └── worker_test.py ├── es │ ├── __init__.py │ ├── blackbox_evaluator.py │ ├── blackbox_evaluator_test.py │ ├── blackbox_learner.py │ ├── blackbox_learner_test.py │ ├── blackbox_optimizers.py │ ├── blackbox_optimizers_test.py │ ├── blackbox_test_utils.py │ ├── es_trainer.py │ ├── es_trainer_lib.py │ ├── gradient_ascent_optimization_algorithms.py │ ├── gradient_ascent_optimization_algorithms_test.py │ ├── inlining │ │ ├── __init__.py │ │ ├── gin_configs │ │ │ ├── blackbox_learner.gin │ │ │ └── inlining.gin │ │ ├── inlining_worker.py │ │ └── inlining_worker_test.py │ ├── policy_utils.py │ ├── policy_utils_test.py │ └── regalloc_trace │ │ ├── __init__.py │ │ ├── gin_configs │ │ ├── blackbox_learner.gin │ │ └── regalloc_trace.gin │ │ ├── regalloc_trace_worker.py │ │ └── regalloc_trace_worker_test.py ├── rl │ ├── __init__.py │ ├── agent_config.py │ ├── agent_config_test.py │ ├── best_trajectory.py │ ├── best_trajectory_test.py │ ├── compilation_runner.py │ ├── compilation_runner_test.py │ ├── constant.py │ ├── constant_value_network.py │ ├── constant_value_network_test.py │ ├── corpus.py │ ├── corpus_test.py │ ├── data_collector.py │ ├── data_collector_test.py │ ├── data_reader.py │ ├── data_reader_test.py │ ├── distributed │ │ ├── __init__.py │ │ ├── agent.py │ │ ├── gin_configs │ │ │ └── regalloc │ │ │ │ ├── common.gin │ │ │ │ ├── ppo_collect.gin │ │ │ │ ├── ppo_eval.gin │ │ │ │ ├── ppo_reverb.gin │ │ │ │ └── ppo_train.gin │ │ ├── learner.py │ │ ├── ppo_collect.py │ │ ├── ppo_collect_lib.py │ │ ├── ppo_eval.py │ │ ├── ppo_eval_lib.py │ │ ├── ppo_reverb_server.py │ │ ├── ppo_reverb_server_lib.py │ │ ├── ppo_train.py │ │ └── ppo_train_lib.py │ ├── env.py │ ├── env_test.py │ ├── feature_ops.py │ ├── feature_ops_test.py │ ├── gin_external_configurables.py │ ├── imitation_learning │ │ ├── __init__.py │ │ ├── generate_bc_trajectories.py │ │ ├── generate_bc_trajectories_lib.py │ │ ├── generate_bc_trajectories_test.py │ │ ├── weighted_bc_trainer.py │ │ ├── weighted_bc_trainer_lib.py │ │ └── weighted_bc_trainer_test.py │ ├── inlining │ │ ├── __init__.py │ │ ├── config.py │ │ ├── env.py │ │ ├── gin_configs │ │ │ ├── actor_behavioral_cloning_nn_agent.gin │ │ │ ├── behavioral_cloning_nn_agent.gin │ │ │ ├── behavioral_cloning_rnn_agent.gin │ │ │ ├── common.gin │ │ │ ├── dqn_nn_agent.gin │ │ │ ├── dqn_rnn_agent.gin │ │ │ ├── imitation_learning.gin │ │ │ └── ppo_nn_agent.gin │ │ ├── imitation_learning_config.py │ │ ├── imitation_learning_runner.py │ │ ├── inlining_runner.py │ │ └── vocab │ │ │ ├── call_argument_setup.buckets │ │ │ ├── call_penalty.buckets │ │ │ ├── callee_basic_block_count.buckets │ │ │ ├── callee_conditionally_executed_blocks.buckets │ │ │ ├── callee_users.buckets │ │ │ ├── caller_basic_block_count.buckets │ │ │ ├── caller_conditionally_executed_blocks.buckets │ │ │ ├── caller_users.buckets │ │ │ ├── callsite_cost.buckets │ │ │ ├── callsite_height.buckets │ │ │ ├── case_cluster_penalty.buckets │ │ │ ├── cold_cc_penalty.buckets │ │ │ ├── constant_args.buckets │ │ │ ├── constant_offset_ptr_args.buckets │ │ │ ├── cost_estimate.buckets │ │ │ ├── dead_blocks.buckets │ │ │ ├── edge_count.buckets │ │ │ ├── indirect_call_penalty.buckets │ │ │ ├── is_multiple_blocks.buckets │ │ │ ├── jump_table_penalty.buckets │ │ │ ├── last_call_to_static_bonus.buckets │ │ │ ├── load_elimination.buckets │ │ │ ├── load_relative_intrinsic.buckets │ │ │ ├── lowered_call_arg_setup.buckets │ │ │ ├── nested_inline_cost_estimate.buckets │ │ │ ├── nested_inlines.buckets │ │ │ ├── node_count.buckets │ │ │ ├── nr_ctant_params.buckets │ │ │ ├── num_loops.buckets │ │ │ ├── simplified_instructions.buckets │ │ │ ├── sroa_losses.buckets │ │ │ ├── sroa_savings.buckets │ │ │ ├── switch_penalty.buckets │ │ │ ├── threshold.buckets │ │ │ └── unsimplified_common_instructions.buckets │ ├── local_data_collector.py │ ├── local_data_collector_test.py │ ├── log_reader.py │ ├── log_reader_test.py │ ├── policy_saver.py │ ├── policy_saver_test.py │ ├── problem_configuration.py │ ├── random_net_distillation.py │ ├── random_net_distillation_test.py │ ├── regalloc │ │ ├── __init__.py │ │ ├── config.py │ │ ├── gin_configs │ │ │ ├── behavioral_cloning_nn_agent.gin │ │ │ ├── common.gin │ │ │ ├── network.gin │ │ │ └── ppo_nn_agent.gin │ │ ├── regalloc_network.py │ │ ├── regalloc_network_test.py │ │ ├── regalloc_runner.py │ │ └── vocab │ │ │ ├── end_bb_freq_by_max.buckets │ │ │ ├── hint_weights_by_max.buckets │ │ │ ├── hottest_bb_freq_by_max.buckets │ │ │ ├── index_to_evict.buckets │ │ │ ├── is_free.buckets │ │ │ ├── is_hint.buckets │ │ │ ├── is_local.buckets │ │ │ ├── liverange_size.buckets │ │ │ ├── mask.buckets │ │ │ ├── max_stage.buckets │ │ │ ├── min_stage.buckets │ │ │ ├── nr_broken_hints.buckets │ │ │ ├── nr_defs_and_uses.buckets │ │ │ ├── nr_rematerializable.buckets │ │ │ ├── nr_urgent.buckets │ │ │ ├── progress.buckets │ │ │ ├── start_bb_freq_by_max.buckets │ │ │ ├── use_def_density.buckets │ │ │ ├── weighed_indvars_by_max.buckets │ │ │ ├── weighed_read_writes_by_max.buckets │ │ │ ├── weighed_reads_by_max.buckets │ │ │ └── weighed_writes_by_max.buckets │ ├── regalloc_priority │ │ ├── __init__.py │ │ ├── config.py │ │ ├── gin_configs │ │ │ ├── common.gin │ │ │ └── ppo_nn_agent.gin │ │ └── regalloc_priority_runner.py │ ├── registry.py │ ├── testdata │ │ └── edge_count.buckets │ ├── train_bc.py │ ├── train_locally.py │ ├── train_locally_lib.py │ ├── trainer.py │ └── trainer_test.py ├── testing │ ├── __init__.py │ ├── corpus_test_utils.py │ └── model_test_utils.py ├── tools │ ├── __init__.py │ ├── combine_tfa_policies.py │ ├── combine_tfa_policies_lib.py │ ├── combine_tfa_policies_lib_test.py │ ├── feature_importance.py │ ├── feature_importance_graphs.py │ ├── feature_importance_utils.py │ ├── feature_importance_utils_test.py │ ├── generate_default_trace.py │ ├── generate_default_trace_lib.py │ ├── generate_default_trace_lib_test.py │ ├── generate_test_model.py │ ├── generate_test_model_test.py │ ├── generate_vocab.py │ ├── merge_best_trajectory.py │ ├── profgen │ │ ├── __init__.py │ │ ├── qualifier.py │ │ ├── toolchain.py │ │ └── toolchain_cmd.py │ └── regalloc_trace │ │ ├── __init__.py │ │ ├── compile_corpus.py │ │ ├── extract_functions.py │ │ ├── extract_functions_lib.py │ │ ├── extract_functions_lib_test.py │ │ ├── group_functions.py │ │ ├── group_functions_lib.py │ │ └── group_functions_lib_test.py └── type_map.py ├── docs ├── adding_features.md ├── contributing.md ├── extensibility.md ├── feature_importance.ipynb ├── inlining-demo │ └── demo.md ├── regalloc-demo │ └── demo.md └── thinlto.md ├── experimental └── docker │ ├── README.md │ └── development.Dockerfile ├── license-header.txt ├── pyproject.toml └── run_tests.sh /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | .vscode 3 | .idea 4 | *.swp 5 | -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/.pylintrc -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/LICENSE -------------------------------------------------------------------------------- /Pipfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/Pipfile -------------------------------------------------------------------------------- /Pipfile.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/Pipfile.lock -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/README.md -------------------------------------------------------------------------------- /check-license.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/check-license.sh -------------------------------------------------------------------------------- /compiler_opt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/__init__.py -------------------------------------------------------------------------------- /compiler_opt/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/__init__.py -------------------------------------------------------------------------------- /compiler_opt/distributed/buffered_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/buffered_scheduler.py -------------------------------------------------------------------------------- /compiler_opt/distributed/buffered_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/buffered_scheduler_test.py -------------------------------------------------------------------------------- /compiler_opt/distributed/local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/local/__init__.py -------------------------------------------------------------------------------- /compiler_opt/distributed/local/local_worker_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/local/local_worker_manager.py -------------------------------------------------------------------------------- /compiler_opt/distributed/local/local_worker_manager_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/local/local_worker_manager_test.py -------------------------------------------------------------------------------- /compiler_opt/distributed/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/worker.py -------------------------------------------------------------------------------- /compiler_opt/distributed/worker_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/worker_manager.py -------------------------------------------------------------------------------- /compiler_opt/distributed/worker_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/distributed/worker_test.py -------------------------------------------------------------------------------- /compiler_opt/es/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/__init__.py -------------------------------------------------------------------------------- /compiler_opt/es/blackbox_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/blackbox_evaluator.py -------------------------------------------------------------------------------- /compiler_opt/es/blackbox_evaluator_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/blackbox_evaluator_test.py -------------------------------------------------------------------------------- /compiler_opt/es/blackbox_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/blackbox_learner.py -------------------------------------------------------------------------------- /compiler_opt/es/blackbox_learner_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/blackbox_learner_test.py -------------------------------------------------------------------------------- /compiler_opt/es/blackbox_optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/blackbox_optimizers.py -------------------------------------------------------------------------------- /compiler_opt/es/blackbox_optimizers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/blackbox_optimizers_test.py -------------------------------------------------------------------------------- /compiler_opt/es/blackbox_test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/blackbox_test_utils.py -------------------------------------------------------------------------------- /compiler_opt/es/es_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/es_trainer.py -------------------------------------------------------------------------------- /compiler_opt/es/es_trainer_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/es_trainer_lib.py -------------------------------------------------------------------------------- /compiler_opt/es/gradient_ascent_optimization_algorithms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/gradient_ascent_optimization_algorithms.py -------------------------------------------------------------------------------- /compiler_opt/es/gradient_ascent_optimization_algorithms_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/gradient_ascent_optimization_algorithms_test.py -------------------------------------------------------------------------------- /compiler_opt/es/inlining/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/inlining/__init__.py -------------------------------------------------------------------------------- /compiler_opt/es/inlining/gin_configs/blackbox_learner.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/inlining/gin_configs/blackbox_learner.gin -------------------------------------------------------------------------------- /compiler_opt/es/inlining/gin_configs/inlining.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/inlining/gin_configs/inlining.gin -------------------------------------------------------------------------------- /compiler_opt/es/inlining/inlining_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/inlining/inlining_worker.py -------------------------------------------------------------------------------- /compiler_opt/es/inlining/inlining_worker_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/inlining/inlining_worker_test.py -------------------------------------------------------------------------------- /compiler_opt/es/policy_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/policy_utils.py -------------------------------------------------------------------------------- /compiler_opt/es/policy_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/policy_utils_test.py -------------------------------------------------------------------------------- /compiler_opt/es/regalloc_trace/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/regalloc_trace/__init__.py -------------------------------------------------------------------------------- /compiler_opt/es/regalloc_trace/gin_configs/blackbox_learner.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/regalloc_trace/gin_configs/blackbox_learner.gin -------------------------------------------------------------------------------- /compiler_opt/es/regalloc_trace/gin_configs/regalloc_trace.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/regalloc_trace/gin_configs/regalloc_trace.gin -------------------------------------------------------------------------------- /compiler_opt/es/regalloc_trace/regalloc_trace_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/regalloc_trace/regalloc_trace_worker.py -------------------------------------------------------------------------------- /compiler_opt/es/regalloc_trace/regalloc_trace_worker_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/es/regalloc_trace/regalloc_trace_worker_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/__init__.py -------------------------------------------------------------------------------- /compiler_opt/rl/agent_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/agent_config.py -------------------------------------------------------------------------------- /compiler_opt/rl/agent_config_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/agent_config_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/best_trajectory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/best_trajectory.py -------------------------------------------------------------------------------- /compiler_opt/rl/best_trajectory_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/best_trajectory_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/compilation_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/compilation_runner.py -------------------------------------------------------------------------------- /compiler_opt/rl/compilation_runner_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/compilation_runner_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/constant.py -------------------------------------------------------------------------------- /compiler_opt/rl/constant_value_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/constant_value_network.py -------------------------------------------------------------------------------- /compiler_opt/rl/constant_value_network_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/constant_value_network_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/corpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/corpus.py -------------------------------------------------------------------------------- /compiler_opt/rl/corpus_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/corpus_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/data_collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/data_collector.py -------------------------------------------------------------------------------- /compiler_opt/rl/data_collector_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/data_collector_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/data_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/data_reader.py -------------------------------------------------------------------------------- /compiler_opt/rl/data_reader_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/data_reader_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/__init__.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/agent.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/gin_configs/regalloc/common.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/gin_configs/regalloc/common.gin -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/gin_configs/regalloc/ppo_collect.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/gin_configs/regalloc/ppo_collect.gin -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/gin_configs/regalloc/ppo_eval.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/gin_configs/regalloc/ppo_eval.gin -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/gin_configs/regalloc/ppo_reverb.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/gin_configs/regalloc/ppo_reverb.gin -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/gin_configs/regalloc/ppo_train.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/gin_configs/regalloc/ppo_train.gin -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/learner.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_collect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_collect.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_collect_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_collect_lib.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_eval.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_eval_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_eval_lib.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_reverb_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_reverb_server.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_reverb_server_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_reverb_server_lib.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_train.py -------------------------------------------------------------------------------- /compiler_opt/rl/distributed/ppo_train_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/distributed/ppo_train_lib.py -------------------------------------------------------------------------------- /compiler_opt/rl/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/env.py -------------------------------------------------------------------------------- /compiler_opt/rl/env_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/env_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/feature_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/feature_ops.py -------------------------------------------------------------------------------- /compiler_opt/rl/feature_ops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/feature_ops_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/gin_external_configurables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/gin_external_configurables.py -------------------------------------------------------------------------------- /compiler_opt/rl/imitation_learning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/imitation_learning/__init__.py -------------------------------------------------------------------------------- /compiler_opt/rl/imitation_learning/generate_bc_trajectories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/imitation_learning/generate_bc_trajectories.py -------------------------------------------------------------------------------- /compiler_opt/rl/imitation_learning/generate_bc_trajectories_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/imitation_learning/generate_bc_trajectories_lib.py -------------------------------------------------------------------------------- /compiler_opt/rl/imitation_learning/generate_bc_trajectories_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/imitation_learning/generate_bc_trajectories_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/imitation_learning/weighted_bc_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/imitation_learning/weighted_bc_trainer.py -------------------------------------------------------------------------------- /compiler_opt/rl/imitation_learning/weighted_bc_trainer_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/imitation_learning/weighted_bc_trainer_lib.py -------------------------------------------------------------------------------- /compiler_opt/rl/imitation_learning/weighted_bc_trainer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/imitation_learning/weighted_bc_trainer_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/__init__.py -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/config.py -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/env.py -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/actor_behavioral_cloning_nn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/actor_behavioral_cloning_nn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/behavioral_cloning_nn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/behavioral_cloning_nn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/behavioral_cloning_rnn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/behavioral_cloning_rnn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/common.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/common.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/dqn_nn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/dqn_nn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/dqn_rnn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/dqn_rnn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/imitation_learning.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/imitation_learning.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/gin_configs/ppo_nn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/gin_configs/ppo_nn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/imitation_learning_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/imitation_learning_config.py -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/imitation_learning_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/imitation_learning_runner.py -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/inlining_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/inlining_runner.py -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/call_argument_setup.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/call_argument_setup.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/call_penalty.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/call_penalty.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/callee_basic_block_count.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/callee_basic_block_count.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/callee_conditionally_executed_blocks.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/callee_conditionally_executed_blocks.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/callee_users.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/callee_users.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/caller_basic_block_count.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/caller_basic_block_count.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/caller_conditionally_executed_blocks.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/caller_conditionally_executed_blocks.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/caller_users.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/caller_users.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/callsite_cost.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/callsite_cost.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/callsite_height.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/callsite_height.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/case_cluster_penalty.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/case_cluster_penalty.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/cold_cc_penalty.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/cold_cc_penalty.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/constant_args.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/constant_args.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/constant_offset_ptr_args.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/constant_offset_ptr_args.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/cost_estimate.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/cost_estimate.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/dead_blocks.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/dead_blocks.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/edge_count.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/edge_count.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/indirect_call_penalty.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/indirect_call_penalty.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/is_multiple_blocks.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/is_multiple_blocks.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/jump_table_penalty.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/jump_table_penalty.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/last_call_to_static_bonus.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/last_call_to_static_bonus.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/load_elimination.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/load_elimination.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/load_relative_intrinsic.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/load_relative_intrinsic.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/lowered_call_arg_setup.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/lowered_call_arg_setup.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/nested_inline_cost_estimate.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/nested_inline_cost_estimate.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/nested_inlines.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/nested_inlines.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/node_count.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/node_count.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/nr_ctant_params.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/nr_ctant_params.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/num_loops.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/num_loops.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/simplified_instructions.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/simplified_instructions.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/sroa_losses.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/sroa_losses.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/sroa_savings.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/sroa_savings.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/switch_penalty.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/switch_penalty.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/threshold.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/threshold.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/inlining/vocab/unsimplified_common_instructions.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/inlining/vocab/unsimplified_common_instructions.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/local_data_collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/local_data_collector.py -------------------------------------------------------------------------------- /compiler_opt/rl/local_data_collector_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/local_data_collector_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/log_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/log_reader.py -------------------------------------------------------------------------------- /compiler_opt/rl/log_reader_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/log_reader_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/policy_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/policy_saver.py -------------------------------------------------------------------------------- /compiler_opt/rl/policy_saver_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/policy_saver_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/problem_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/problem_configuration.py -------------------------------------------------------------------------------- /compiler_opt/rl/random_net_distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/random_net_distillation.py -------------------------------------------------------------------------------- /compiler_opt/rl/random_net_distillation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/random_net_distillation_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/__init__.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/config.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/gin_configs/behavioral_cloning_nn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/gin_configs/behavioral_cloning_nn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/gin_configs/common.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/gin_configs/common.gin -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/gin_configs/network.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/gin_configs/network.gin -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/gin_configs/ppo_nn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/gin_configs/ppo_nn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/regalloc_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/regalloc_network.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/regalloc_network_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/regalloc_network_test.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/regalloc_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/regalloc_runner.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/end_bb_freq_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/end_bb_freq_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/hint_weights_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/hint_weights_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/hottest_bb_freq_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/hottest_bb_freq_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/index_to_evict.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/index_to_evict.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/is_free.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/is_free.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/is_hint.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/is_hint.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/is_local.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/is_local.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/liverange_size.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/liverange_size.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/mask.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/mask.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/max_stage.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/max_stage.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/min_stage.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/min_stage.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/nr_broken_hints.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/nr_broken_hints.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/nr_defs_and_uses.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/nr_defs_and_uses.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/nr_rematerializable.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/nr_rematerializable.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/nr_urgent.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/nr_urgent.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/progress.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/progress.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/start_bb_freq_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/start_bb_freq_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/use_def_density.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/use_def_density.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/weighed_indvars_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/weighed_indvars_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/weighed_read_writes_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/weighed_read_writes_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/weighed_reads_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/weighed_reads_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc/vocab/weighed_writes_by_max.buckets: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc/vocab/weighed_writes_by_max.buckets -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc_priority/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc_priority/__init__.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc_priority/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc_priority/config.py -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc_priority/gin_configs/common.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc_priority/gin_configs/common.gin -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc_priority/gin_configs/ppo_nn_agent.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc_priority/gin_configs/ppo_nn_agent.gin -------------------------------------------------------------------------------- /compiler_opt/rl/regalloc_priority/regalloc_priority_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/regalloc_priority/regalloc_priority_runner.py -------------------------------------------------------------------------------- /compiler_opt/rl/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/registry.py -------------------------------------------------------------------------------- /compiler_opt/rl/testdata/edge_count.buckets: -------------------------------------------------------------------------------- 1 | 2 2 | 2 3 | 2 4 | 3 5 | 4 6 | 5 7 | 8 8 | 16 9 | 51 10 | -------------------------------------------------------------------------------- /compiler_opt/rl/train_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/train_bc.py -------------------------------------------------------------------------------- /compiler_opt/rl/train_locally.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/train_locally.py -------------------------------------------------------------------------------- /compiler_opt/rl/train_locally_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/train_locally_lib.py -------------------------------------------------------------------------------- /compiler_opt/rl/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/trainer.py -------------------------------------------------------------------------------- /compiler_opt/rl/trainer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/rl/trainer_test.py -------------------------------------------------------------------------------- /compiler_opt/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/testing/__init__.py -------------------------------------------------------------------------------- /compiler_opt/testing/corpus_test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/testing/corpus_test_utils.py -------------------------------------------------------------------------------- /compiler_opt/testing/model_test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/testing/model_test_utils.py -------------------------------------------------------------------------------- /compiler_opt/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/__init__.py -------------------------------------------------------------------------------- /compiler_opt/tools/combine_tfa_policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/combine_tfa_policies.py -------------------------------------------------------------------------------- /compiler_opt/tools/combine_tfa_policies_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/combine_tfa_policies_lib.py -------------------------------------------------------------------------------- /compiler_opt/tools/combine_tfa_policies_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/combine_tfa_policies_lib_test.py -------------------------------------------------------------------------------- /compiler_opt/tools/feature_importance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/feature_importance.py -------------------------------------------------------------------------------- /compiler_opt/tools/feature_importance_graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/feature_importance_graphs.py -------------------------------------------------------------------------------- /compiler_opt/tools/feature_importance_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/feature_importance_utils.py -------------------------------------------------------------------------------- /compiler_opt/tools/feature_importance_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/feature_importance_utils_test.py -------------------------------------------------------------------------------- /compiler_opt/tools/generate_default_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/generate_default_trace.py -------------------------------------------------------------------------------- /compiler_opt/tools/generate_default_trace_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/generate_default_trace_lib.py -------------------------------------------------------------------------------- /compiler_opt/tools/generate_default_trace_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/generate_default_trace_lib_test.py -------------------------------------------------------------------------------- /compiler_opt/tools/generate_test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/generate_test_model.py -------------------------------------------------------------------------------- /compiler_opt/tools/generate_test_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/generate_test_model_test.py -------------------------------------------------------------------------------- /compiler_opt/tools/generate_vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/generate_vocab.py -------------------------------------------------------------------------------- /compiler_opt/tools/merge_best_trajectory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/merge_best_trajectory.py -------------------------------------------------------------------------------- /compiler_opt/tools/profgen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/profgen/__init__.py -------------------------------------------------------------------------------- /compiler_opt/tools/profgen/qualifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/profgen/qualifier.py -------------------------------------------------------------------------------- /compiler_opt/tools/profgen/toolchain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/profgen/toolchain.py -------------------------------------------------------------------------------- /compiler_opt/tools/profgen/toolchain_cmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/profgen/toolchain_cmd.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/__init__.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/compile_corpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/compile_corpus.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/extract_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/extract_functions.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/extract_functions_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/extract_functions_lib.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/extract_functions_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/extract_functions_lib_test.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/group_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/group_functions.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/group_functions_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/group_functions_lib.py -------------------------------------------------------------------------------- /compiler_opt/tools/regalloc_trace/group_functions_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/tools/regalloc_trace/group_functions_lib_test.py -------------------------------------------------------------------------------- /compiler_opt/type_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/compiler_opt/type_map.py -------------------------------------------------------------------------------- /docs/adding_features.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/docs/adding_features.md -------------------------------------------------------------------------------- /docs/contributing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/docs/contributing.md -------------------------------------------------------------------------------- /docs/extensibility.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/docs/extensibility.md -------------------------------------------------------------------------------- /docs/feature_importance.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/docs/feature_importance.ipynb -------------------------------------------------------------------------------- /docs/inlining-demo/demo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/docs/inlining-demo/demo.md -------------------------------------------------------------------------------- /docs/regalloc-demo/demo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/docs/regalloc-demo/demo.md -------------------------------------------------------------------------------- /docs/thinlto.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/docs/thinlto.md -------------------------------------------------------------------------------- /experimental/docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/experimental/docker/README.md -------------------------------------------------------------------------------- /experimental/docker/development.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/experimental/docker/development.Dockerfile -------------------------------------------------------------------------------- /license-header.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/license-header.txt -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/pyproject.toml -------------------------------------------------------------------------------- /run_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/ml-compiler-opt/HEAD/run_tests.sh --------------------------------------------------------------------------------