From b6406a43b48b7803f3efdbc235b1fbb5da782449 Mon Sep 17 00:00:00 2001 From: mlevesquedion Date: Fri, 26 Apr 2024 14:57:05 -0700 Subject: [PATCH] Fix extraDeclarations for AllGather (#2262) 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(); }];