Fix compilation errors in ReshapeOpCreate refactoring

- Remove .getResult() call on Value at line 1047
- Cast Value to RankedTensorType before getShape() at line 6105
- Remove unused variable post_reshape at line 21655
- Call getDefiningOp() on Value before setShardings at lines 22797, 22809, 23237
- Remove unused variables newLhsType, newRhsType at lines 31319-31322
- Remove unused variable newInputTy at line 31431

Co-authored-by: avik-pal <30564094+avik-pal@users.noreply.github.com>

Author: Copilot

src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp

@@ -1044,7 +1044,7 @@ struct ReshapeExtend final
 
     // Then create a new extend operation on the reshaped data
     auto newExtendOp = enzymexla::ExtendOp::create(rewriter, op.getLoc(),
-                                                   newReshapeOp.getResult(),
+                                                   newReshapeOp,
                                                    lhs, rhs, newExtendDim);
 
     // Replace the original reshape op with the new extend operation
@@ -6102,7 +6102,7 @@ struct ConcatToBroadcast final
           rewriter, op.getLoc(), op->getOperand(0), reshaped);
 
       SmallVector<int64_t> bcast;
-      for (auto en : llvm::enumerate(reshapeVal.getType().getShape())) {
+      for (auto en : llvm::enumerate(cast<RankedTensorType>(reshapeVal.getType()).getShape())) {
         bcast.push_back(en.index());
       }
       reshaped[op.getDimension()] = op->getNumOperands();
@@ -21652,8 +21652,6 @@ struct SumToConv : public SumToReductionBase<ST, SumToConv<ST>> {
     if (conv.getType() != pre_reshape) {
       SmallVector<int64_t> post_shape = llvm::to_vector(pre_reshape.getShape());
       post_shape[reshapeOffsetDim] -= (lastidx - startidx);
-      RankedTensorType post_reshape =
-          RankedTensorType::get(post_shape, pre_reshape.getElementType());
       conv = stablehlo::ReshapeOpCreate(rewriter, input.getLoc(),
                                         conv, post_shape);
     }
@@ -22794,7 +22792,7 @@ struct RecognizeWrap
           auto reshape = stablehlo::ReshapeOpCreate(
               rewriter, concat.getLoc(), wrap, newShape);
           if (auto shard = sdy::getShardingPerValue(rs0)) {
-            sdy::setShardings(reshape, shard);
+            sdy::setShardings(reshape.getDefiningOp(), shard);
           }
           toConcat.push_back(reshape);
           for (int j = i + 1; j < operands.size(); j++)
@@ -22806,7 +22804,7 @@ struct RecognizeWrap
             rewriter.replaceOpWithNewOp<stablehlo::ConcatenateOp>(
                 concat, toConcat, concatDim);
             if (shard) {
-              sdy::setShardings(reshape, shard);
+              sdy::setShardings(reshape.getDefiningOp(), shard);
             }
           }
           return success();
@@ -23234,7 +23232,7 @@ struct RecognizeExtend
         auto reshape = stablehlo::ReshapeOpCreate(
             rewriter, concat.getLoc(), extend, shape);
         if (auto shard = sdy::getShardingPerValue(concat)) {
-          sdy::setShardings(reshape, shard);
+          sdy::setShardings(reshape.getDefiningOp(), shard);
         }
         finish(reshape);
         return success();
@@ -31316,11 +31314,6 @@ struct DotGeneralInsertDimContractionSimplification final
     }
 
     // Create reshaped operands. This will be cleaned up later
-    auto newLhsType =
-        RankedTensorType::get(newLhsShape, lhsType.getElementType());
-    auto newRhsType =
-        RankedTensorType::get(newRhsShape, rhsType.getElementType());
-
     Value newLhs =
         stablehlo::ReshapeOpCreate(rewriter, op.getLoc(), lhs, newLhsShape);
     Value newRhs =
@@ -31428,8 +31421,6 @@ struct DeleteDimsBroadcast final
     }
 
     // Create the reshape on the input
-    auto newInputTy =
-        RankedTensorType::get(newInputShape, bcastInputTy.getElementType());
     auto reshapeInput = stablehlo::ReshapeOpCreate(
         rewriter, op.getLoc(), bcastOp.getOperand(), newInputShape);