diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-09-06 20:59:08 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-06 21:03:25 -0700 |
| commit | 27ce2a4ac956941ba8a0b9aaaa77acc0aa861fef (patch) | |
| tree | 23f9703c2948b7b464d93b44f701d5e1dcd0f311 /tensorflow/compiler/xla/service/hlo_verifier.h | |
| parent | af41a029c8def570c409167326f59a5222452f40 (diff) | |
[XLA] Rip CheckFusionNode() out of instruction, and move it into the HLO verifier instead.
CheckFusionNode() is linear in the size of the fusion node, and was called once per Fuse(), leading to run-time quadratic in the fusion node's size.
PiperOrigin-RevId: 167812735
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_verifier.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_verifier.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_verifier.h b/tensorflow/compiler/xla/service/hlo_verifier.h index bc6800dae5..e35a7f3642 100644 --- a/tensorflow/compiler/xla/service/hlo_verifier.h +++ b/tensorflow/compiler/xla/service/hlo_verifier.h @@ -34,6 +34,9 @@ class HloVerifier : public HloPassInterface { StatusOr<bool> Run(HloModule* module) override; private: + // CHECKs various invariants of a fusion instruction. + Status CheckFusionInstruction(HloInstruction* fusion) const; + // Returns the size of a Shape in bytes. const std::function<int64(const Shape&)> shape_size_fn_; }; |