diff options
author | 2018-08-24 17:53:34 -0700 | |
---|---|---|
committer | 2018-08-24 17:56:59 -0700 | |
commit | 481266fa991ca2b08221214f535c6a8c546e2350 (patch) | |
tree | 343e79e0f0dc852886b33ebeb879b6efa9984779 /tensorflow/compiler/xla/service/BUILD | |
parent | be7a5e4b6e50842bc3c841daaa8dadadc793dd5f (diff) |
Run AddSpecialCaseCopies in HloRematerialization.
RemoveUnnecessaryCopies which runs in rematerialization to take advantage of scheduling can sometimes remove copies which are needed to non-interference reasons. This requires running AddSpecialCaseCopies to add them back in. Furthermore, the schedule needs to be updated to account for the changes to the module, so add an UpdateSchedule function which can patch up a schedule in light a limited set of transformations to the module (addition and deletion of instructions).
PiperOrigin-RevId: 210186375
Diffstat (limited to 'tensorflow/compiler/xla/service/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/service/BUILD | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index aa826aa770..47d376c8ac 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -1152,6 +1152,7 @@ cc_library( "//tensorflow/compiler/xla:types", "//tensorflow/compiler/xla:util", "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", ], ) @@ -1159,17 +1160,18 @@ tf_cc_test( name = "hlo_scheduling_test", srcs = ["hlo_scheduling_test.cc"], deps = [ - ":buffer_value", ":heap_simulator", ":hlo", + ":hlo_dce", ":hlo_ordering", + ":hlo_parser", ":hlo_scheduling", "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:types", "//tensorflow/compiler/xla:xla_data_proto", - "//tensorflow/compiler/xla/service:hlo_parser", "//tensorflow/compiler/xla/tests:hlo_test_base", "//tensorflow/compiler/xla/tests:xla_internal_test_main", + "//tensorflow/core:test", ], ) @@ -2320,6 +2322,7 @@ cc_library( "//tensorflow/compiler/xla:types", "//tensorflow/compiler/xla:util", "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/strings", ], |