Skip to content

Commit 6332087

Browse files
authored
move regalloc_network to regalloc/regalloc_network (#64)
* move regalloc_network to regalloc/regalloc_network * add back import in bc.gin
1 parent 8ff4f65 commit 6332087

File tree

5 files changed

+4
-4
lines changed

5 files changed

+4
-4
lines changed

compiler_opt/rl/regalloc/gin_configs/behavioral_cloning_nn_agent.gin

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ import gin.tf.external_configurables
22
import compiler_opt.rl.constant
33
import compiler_opt.rl.gin_external_configurables
44
import compiler_opt.rl.regalloc.config
5-
import compiler_opt.rl.regalloc_network
5+
import compiler_opt.rl.regalloc.regalloc_network
66
import tf_agents.agents.behavioral_cloning.behavioral_cloning_agent
77
import tf_agents.networks.actor_distribution_network
88

compiler_opt/rl/regalloc/gin_configs/network.gin

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import compiler_opt.rl.gin_external_configurables
22
import compiler_opt.rl.regalloc.config
3-
import compiler_opt.rl.regalloc_network
3+
import compiler_opt.rl.regalloc.regalloc_network
44

55
regalloc.config.get_observation_processing_layer_creator.quantile_file_dir='compiler_opt/rl/regalloc/vocab'
66
regalloc.config.get_observation_processing_layer_creator.with_sqrt = False

compiler_opt/rl/regalloc/gin_configs/ppo_nn_agent.gin

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ import compiler_opt.rl.constant
33
import compiler_opt.rl.constant_value_network
44
import compiler_opt.rl.gin_external_configurables
55
import compiler_opt.rl.regalloc.config
6-
import compiler_opt.rl.regalloc_network
6+
import compiler_opt.rl.regalloc.regalloc_network
77
import tf_agents.agents.ppo.ppo_agent
88
import tf_agents.networks.actor_distribution_network
99

compiler_opt/rl/regalloc_network_test.py renamed to compiler_opt/rl/regalloc/regalloc_network_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818
import tensorflow_probability as tfp
1919
from tf_agents.specs import tensor_spec
2020

21-
from compiler_opt.rl import regalloc_network
2221
from compiler_opt.rl.regalloc import config
22+
from compiler_opt.rl.regalloc import regalloc_network
2323

2424

2525
def _observation_processing_layer(obs_spec):

0 commit comments

Comments
 (0)