fix operand

EnzymeAD · Feb 8, 2025 · c1a0a26 · c1a0a26
1 parent d8940ba
commit c1a0a26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp b/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp
@@ -5771,7 +5771,7 @@ struct GatherSimplify final : OpRewritePattern<mlir::stablehlo::GatherOp> {
 
     {
       DenseIntElementsAttr operandVals;
-      if (matchPattern(operand, m_Constant(&operandVals))) {
+      if (matchPattern(op.getOperand(), m_Constant(&operandVals))) {
         auto out = stablehlo::gatherOp(
             stablehlo::constantOp(operandVals),
             stablehlo::constantOp(startIndicesCst),