Skip to content

Commit

Permalink
[XLA:GPU] Disable ALL fusion passes in xla_hlo_profile_test.
Browse files Browse the repository at this point in the history
Rename the "fusion merger" pass to "fusion_merger" such that it can be disabled.
This is in preparation of handling fused and unfused ops more consistently.

PiperOrigin-RevId: 253969358
  • Loading branch information
thomasjoerg authored and tensorflower-gardener committed Jun 19, 2019
1 parent 1d589da commit a98b781
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion tensorflow/compiler/xla/service/gpu/fusion_merger.h
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ namespace gpu {
//
class FusionMerger : public HloModulePass {
public:
absl::string_view name() const override { return "fusion merger"; }
absl::string_view name() const override { return "fusion_merger"; }

StatusOr<bool> Run(HloModule* module) override;
};
Expand Down
3 changes: 2 additions & 1 deletion tensorflow/compiler/xla/tests/xla_hlo_profile_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -376,7 +376,8 @@ static std::pair<int, char**> AddXlaHloProfileFlag(int argc, char** argv) {
// pass, otherwise a while loop is transformed and we could not match the
// original name in the ProfileWhileComputation test.
new_argv[argc + 1] = strdup(
"--xla_disable_hlo_passes=fusion,while-loop-invariant-code-motion");
"--xla_disable_hlo_passes=fusion,fusion_merger,multi_output_fusion,"
"while-loop-invariant-code-motion");
return {argc + 2, new_argv};
}

Expand Down

0 comments on commit a98b781

Please sign in to comment.