diff options
author | 2018-09-06 08:56:46 -0700 | |
---|---|---|
committer | 2018-09-06 09:01:16 -0700 | |
commit | a41e270641f0613413e1929c9010f32882b4d26b (patch) | |
tree | e2b5bee9504f8e6319b1d40c8d56fb8f7fbf4cce /tensorflow/compiler/xla/service/BUILD | |
parent | 35f28c57da8aad4a79503db955b11fed63b1fe34 (diff) |
Add HloSchedule to HloModule.
Add HloSchedule as a field on HloModule. This will enable scheduling to be a normal HLO pass and enable some passes such as copy insertion to more easily use tighter instruction live ranges based on the schedule. This change required adding HloSchedule to the "hlo" library because of circular dependencies.
Nothing except for tests actually sets the schedule at the moment, but follow up cls will add a scheduling pass which will do so.
PiperOrigin-RevId: 211815293
Diffstat (limited to 'tensorflow/compiler/xla/service/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/service/BUILD | 30 |
1 files changed, 7 insertions, 23 deletions
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index ab86dce510..b8ee6a093e 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -291,6 +291,7 @@ cc_library( "hlo_instructions.cc", "hlo_module.cc", "hlo_opcode.cc", + "hlo_schedule.cc", "hlo_sharding.cc", ], hdrs = [ @@ -303,6 +304,7 @@ cc_library( "hlo_instructions.h", "hlo_module.h", "hlo_opcode.h", + "hlo_schedule.h", "hlo_sharding.h", ], deps = [ @@ -331,6 +333,8 @@ cc_library( "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/memory", "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/types:optional", "@com_google_absl//absl/types:span", ], ) @@ -1037,7 +1041,6 @@ tf_cc_test( ":flatten_call_graph", ":hlo", ":hlo_ordering", - ":hlo_schedule", ":hlo_scheduling", "//tensorflow/compiler/xla:literal", "//tensorflow/compiler/xla:shape_util", @@ -1065,7 +1068,6 @@ cc_library( ":hlo", ":hlo_dataflow_analysis", ":hlo_proto", - ":hlo_schedule", ":hlo_value", "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:status_macros", @@ -1086,7 +1088,6 @@ tf_cc_test( ":hlo", ":hlo_dataflow_analysis", ":hlo_ordering", - ":hlo_schedule", ":hlo_scheduling", "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:types", @@ -1108,7 +1109,6 @@ cc_library( ":hlo", ":hlo_ordering", ":hlo_proto", - ":hlo_schedule", ":tuple_points_to_analysis", "//tensorflow/compiler/xla:statusor", "//tensorflow/compiler/xla:util", @@ -1177,22 +1177,6 @@ cc_library( ], ) -cc_library( - name = "hlo_schedule", - srcs = ["hlo_schedule.cc"], - hdrs = ["hlo_schedule.h"], - deps = [ - ":hlo", - "//tensorflow/compiler/xla:status", - "//tensorflow/compiler/xla:status_macros", - "//tensorflow/compiler/xla:util", - "//tensorflow/core:lib_internal", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/types:span", - ], -) - tf_cc_test( name = "hlo_schedule_test", srcs = ["hlo_schedule_test.cc"], @@ -1202,7 +1186,6 @@ tf_cc_test( ":hlo_dce", ":hlo_ordering", ":hlo_parser", - ":hlo_schedule", ":hlo_scheduling", "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:types", @@ -1222,7 +1205,6 @@ cc_library( ":heap_simulator", ":hlo", ":hlo_ordering", - ":hlo_schedule", ":logical_buffer", ":tuple_points_to_analysis", "//tensorflow/compiler/xla:shape_util", @@ -1969,6 +1951,8 @@ tf_cc_test( srcs = ["hlo_module_test.cc"], deps = [ ":hlo", + ":hlo_matchers", + ":hlo_parser", "//tensorflow/compiler/xla:literal", "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:test", @@ -1977,6 +1961,7 @@ tf_cc_test( "//tensorflow/compiler/xla/tests:hlo_test_base", "//tensorflow/compiler/xla/tests:xla_internal_test_main", "//tensorflow/core:lib", + "//tensorflow/core:test", "@com_google_absl//absl/memory", "@com_google_absl//absl/types:span", ], @@ -2413,7 +2398,6 @@ cc_library( ":hlo", ":hlo_dce", ":hlo_ordering", - ":hlo_schedule", ":hlo_scheduling", ":logical_buffer", ":tuple_points_to_analysis", |