Fix extraDeclarations for AllGather (#2262)

Missing the common declarations.
openxla · Apr 26, 2024 · b6406a4 · b6406a4
1 parent faf995a
commit b6406a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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();
   }];