More info on the lockstep failure errors

Author: wsmoses

src/enzyme_ad/jax/Implementations/StableHLOAutoDiffOpInterfaceImpl.cpp

@@ -467,10 +467,10 @@ class AutoDiffWhileFwd
         auto idx = arg.getArgNumber();
         if (resultPositionsToShadow.count(idx)) {
           if (gutils->isConstantValue(arg)) {
-            nb->insertArgument(
-                curidx,
-                cast<AutoDiffTypeInterface>(arg.getType()).getShadowType(gutils->width),
-                op.getLoc());
+            nb->insertArgument(curidx,
+                               cast<AutoDiffTypeInterface>(arg.getType())
+                                   .getShadowType(gutils->width),
+                               op.getLoc());
           }
           curidx++;
         }

src/enzyme_ad/jax/Passes/AffineToStableHLORaising.cpp

@@ -1668,7 +1668,7 @@ static LogicalResult tryRaisingLockStepForOpToStableHLO(
   if (pc.options.dump_failed_lockstep) {
     llvm::errs() << " failed lockstep of for raise: " << *forOp << "\n";
   }
-  return forOp.emitError("Not lockstep executable");
+  return forOp.emitError("Not lockstep executable") << *forOp;
 }
 
 static LogicalResult