diff options
| author |  Mark Heffernan <meheff@google.com> | 2017-02-21 18:17:54 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-21 18:38:46 -0800 |
| commit | 7817ac8055ca328e5bf902677f335502eb0da926 (patch) | |
| tree | c058884aea29e809622176f690a8edd0feb1f176 /tensorflow/compiler/xla/service/local_service.cc | |
| parent | f3405c2d73196e409041d52bbf30748b2a64493b (diff) | |
[XLA] Properly version outfeed and send operations in UserComputation.
Previously outfeed and send operations were unconditionally emitted during
UserComputation lowering even if the outfeed/send was not in the requested
version (computation snapshot). This CL versions these operations.
Also, opportunistically improve logging in UserComputation, Service, and
ComputationTracker which was used to root cause the underlying bug.
Change: 148170893
Diffstat (limited to 'tensorflow/compiler/xla/service/local_service.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/local_service.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/local_service.cc b/tensorflow/compiler/xla/service/local_service.cc index 402cc2b615..02abd0d8a9 100644 --- a/tensorflow/compiler/xla/service/local_service.cc +++ b/tensorflow/compiler/xla/service/local_service.cc @@ -223,7 +223,7 @@ LocalService::CompileAheadOfTime( TF_ASSIGN_OR_RETURN(std::unique_ptr<HloModule> hlo_module, computation_tracker_.BuildHloModule( versioned_handle, - /*include_unused_parameters=*/true)); + /*include_unreachable_instructions=*/true)); hlo_modules.push_back(std::move(hlo_module)); TF_ASSIGN_OR_RETURN( |