From c3b54a8a1df84dd36b371ee37e88bcbbaee505b1 Mon Sep 17 00:00:00 2001
From: Praveen Narayanan <pravnar@google.com>
Date: Wed, 20 Nov 2024 01:11:52 -0800
Subject: [PATCH] Add a pattern matcher for ragged dot HLO.

PiperOrigin-RevId: 698297679
---
 xla/service/pattern_matcher.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/xla/service/pattern_matcher.h b/xla/service/pattern_matcher.h
index eacd17ed0bfb4..f5267bc688e78 100644
--- a/xla/service/pattern_matcher.h
+++ b/xla/service/pattern_matcher.h
@@ -2816,6 +2816,7 @@ XLA_COMMUTATIVE_BINOP_PATTERN(Xor)
         .WithOperand(2, std::forward<Arg2>(arg2));                     \
   }
 XLA_TERNOP_PATTERN(Clamp);
+XLA_TERNOP_PATTERN(RaggedDot);
 XLA_TERNOP_PATTERN(Select);
 XLA_TERNOP_PATTERN(SelectAndScatter);
 #undef XLA_TERNOP_PATTERN