diff options
author | 2017-07-19 10:59:15 -0700 | |
---|---|---|
committer | 2017-07-19 11:05:10 -0700 | |
commit | c2182e05ebb1a2dea8a554eb1644dea7e593a8b1 (patch) | |
tree | 82331cdc775f36c84a185d2bfe724fb22b3207d5 /tensorflow/compiler/xla/service | |
parent | cae9d0f3fe6b1bc37f0e5a69765239f89a05ce6e (diff) |
[XLA] Move the flag from stream_assignment_flags into DebugOptions
PiperOrigin-RevId: 162506964
Diffstat (limited to 'tensorflow/compiler/xla/service')
-rw-r--r-- | tensorflow/compiler/xla/service/gpu/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/compiler/xla/service/gpu/stream_assignment.cc | 8 |
2 files changed, 4 insertions, 5 deletions
diff --git a/tensorflow/compiler/xla/service/gpu/BUILD b/tensorflow/compiler/xla/service/gpu/BUILD index 6809099695..fa95e23499 100644 --- a/tensorflow/compiler/xla/service/gpu/BUILD +++ b/tensorflow/compiler/xla/service/gpu/BUILD @@ -68,7 +68,6 @@ cc_library( deps = [ ":ir_emission_utils", "//tensorflow/compiler/xla:util", - "//tensorflow/compiler/xla/legacy_flags:stream_assignment_flags", "//tensorflow/compiler/xla/service:hlo", "//tensorflow/compiler/xla/service:hlo_reachability", "//tensorflow/core:lib", diff --git a/tensorflow/compiler/xla/service/gpu/stream_assignment.cc b/tensorflow/compiler/xla/service/gpu/stream_assignment.cc index a304decc49..e4cfc6999f 100644 --- a/tensorflow/compiler/xla/service/gpu/stream_assignment.cc +++ b/tensorflow/compiler/xla/service/gpu/stream_assignment.cc @@ -15,7 +15,6 @@ limitations under the License. #include "tensorflow/compiler/xla/service/gpu/stream_assignment.h" -#include "tensorflow/compiler/xla/legacy_flags/stream_assignment_flags.h" #include "tensorflow/compiler/xla/map_util.h" #include "tensorflow/compiler/xla/ptr_util.h" #include "tensorflow/compiler/xla/service/gpu/ir_emission_utils.h" @@ -66,9 +65,10 @@ int ComputeStreamToAssign( return -1; } - legacy_flags::StreamAssignmentFlags* flags = - legacy_flags::GetStreamAssignmentFlags(); - if (flags->xla_gpu_disable_multi_streaming) { + if (hlo.GetModule() + ->config() + .debug_options() + .xla_gpu_disable_multi_streaming()) { return 0; } |