diff options
author | Mark Heffernan <meheff@google.com> | 2018-09-19 08:12:29 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 08:16:42 -0700 |
commit | f8655c08cfe3bd99ec1703211e1c9154a14a6150 (patch) | |
tree | 90bf5c29d3a1f77764c2f2392c4b0564b490c995 /tensorflow/compiler/xla/service/copy_insertion.h | |
parent | e1db78697b05be673562fe2b1c9a995d25a71d4c (diff) |
Add interface for HLO passes which run on HloModuleGroup.
Derive HloModulePass and HloModuleGroupPass from HloPassInterface which run module-scoped and module-group-scoped respectively. Replace all existing uses of HloPassInterface with HloModulePass because all existing passes are module-scoped. Also rewrite HloPassPipeline to support both module-scoped and module-group-scoped passes.
PiperOrigin-RevId: 213629604
Diffstat (limited to 'tensorflow/compiler/xla/service/copy_insertion.h')
-rw-r--r-- | tensorflow/compiler/xla/service/copy_insertion.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/copy_insertion.h b/tensorflow/compiler/xla/service/copy_insertion.h index d308f6bc84..c097089e30 100644 --- a/tensorflow/compiler/xla/service/copy_insertion.h +++ b/tensorflow/compiler/xla/service/copy_insertion.h @@ -43,7 +43,7 @@ namespace xla { // (3) The buffer set of the root instruction of the entry computation must be // unambiguous and distinct. That is, InstructionAliasSet::IsAmbiguous and // InstructionAliasSet::IsDistinct return true. -class CopyInsertion : public HloPassInterface { +class CopyInsertion : public HloModulePass { public: absl::string_view name() const override { return "copy-insertion"; } |