Add a pattern matcher for ragged dot HLO.

PiperOrigin-RevId: 689135409
openxla · Nov 20, 2024 · b926c13 · b926c13
1 parent 74da9de
commit b926c13
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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