diff options
author | 2018-09-06 16:40:47 -0700 | |
---|---|---|
committer | 2018-09-06 16:46:11 -0700 | |
commit | c4df798540b83026ccc74d69da38960e43af8f55 (patch) | |
tree | 73945b9307eabdfef9604a0a27a9ecae8fa2cc37 /tensorflow/compiler/xla/service/hlo_cost_analysis.cc | |
parent | 7caba396ba81a0a19efd92a01aa7a3b695e3009b (diff) |
[XLA] Handle kDomain in HloCostAnalysis.
PiperOrigin-RevId: 211891325
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_cost_analysis.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_cost_analysis.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_cost_analysis.cc b/tensorflow/compiler/xla/service/hlo_cost_analysis.cc index 939b5114c3..8b4eaad82e 100644 --- a/tensorflow/compiler/xla/service/hlo_cost_analysis.cc +++ b/tensorflow/compiler/xla/service/hlo_cost_analysis.cc @@ -227,6 +227,14 @@ Status HloCostAnalysis::HandleCopy(const HloInstruction*) { return Status::OK(); } +Status HloCostAnalysis::HandleDomain(const HloInstruction* domain) { + // Domain does not have any computation or data transfer. + current_should_compute_bottleneck_time_ = false; + current_properties_[kBytesAccessedKey] = 0; + current_properties_[kOptimalSecondsKey] = 0; + return Status::OK(); +} + Status HloCostAnalysis::HandleDot(const HloInstruction* dot) { const Shape& lhs_shape = dot->operand(0)->shape(); const Shape& rhs_shape = dot->operand(1)->shape(); |