From e761103744b40474bd85caa58d02fbfcecc118f2 Mon Sep 17 00:00:00 2001
From: Yuanzhong Xu <yuanzx@google.com>
Date: Thu, 21 May 2020 22:24:50 -0700
Subject: [PATCH] [XLA] Add AllGather to the HLO matchers.

PiperOrigin-RevId: 312797129
Change-Id: I6a862f34a3b2331d99fe0bd242e21f26da7ed99e
---
 tensorflow/compiler/xla/service/hlo_matchers.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/compiler/xla/service/hlo_matchers.h b/tensorflow/compiler/xla/service/hlo_matchers.h
index ec048bef9e8..cb1b1d0dae4 100644
--- a/tensorflow/compiler/xla/service/hlo_matchers.h
+++ b/tensorflow/compiler/xla/service/hlo_matchers.h
@@ -203,6 +203,7 @@ HLO_MATCHER(Abs);
 HLO_MATCHER(Add);
 HLO_MATCHER(AddDependency);
 HLO_MATCHER(AfterAll);
+HLO_MATCHER(AllGather);
 HLO_MATCHER(AllReduce);
 HLO_MATCHER(AllToAll);
 HLO_MATCHER(And);