Fix extraDeclarations for AllGather

Missing the common declarations.
openxla · Apr 26, 2024 · b350c05 · b350c05
1 parent faf995a
commit b350c05
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();
   }];