From b350c05400283b7235e1ec93ff71f9e58b3871d2 Mon Sep 17 00:00:00 2001
From: Michael Levesque-Dion <mlevesquedion@google.com>
Date: Fri, 26 Apr 2024 13:45:07 -0700
Subject: [PATCH] Fix extraDeclarations for AllGather

Missing the common declarations.
---
 stablehlo/dialect/StablehloOps.td | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/stablehlo/dialect/StablehloOps.td b/stablehlo/dialect/StablehloOps.td
index a4d2ec4e4e..25211175f9 100644
--- a/stablehlo/dialect/StablehloOps.td
+++ b/stablehlo/dialect/StablehloOps.td
@@ -1349,7 +1349,7 @@ def StableHLO_AllGatherOp : StableHLO_Op<"all_gather",
   let results = (outs HLO_Tensor);
   let hasVerifier = 1;
 
-  let extraClassDeclaration = [{
+  let extraClassDeclaration = commonClassDeclaration # [{
     /// Interface method for ConditionallySpeculatable.
     mlir::Speculation::Speculatability getSpeculatability();
   }];