commit b6406a43b48b7803f3efdbc235b1fbb5da782449
author mlevesquedion <mlevesquedion@google.com> Fri Apr 26 14:57:05 2024 -0700
committer GitHub <noreply@github.com> Fri Apr 26 14:57:05 2024 -0700
tree 6df2091a8fe8810825867a6e05211ac6b648eec7
parent faf995aec3aab212d79a7a5a18f26a4b33095e78

Fix extraDeclarations for AllGather (#2262)

Missing the common declarations.

stablehlo/dialect/StablehloOps.td

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