diff --git a/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp b/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp index ee49fe332..4fab60d25 100644 --- a/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp +++ b/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp @@ -21576,6 +21576,8 @@ struct ConcatenateBroadcastInDim if (operands.size() == 1) return failure(); + llvm::errs() << op->getParentOfType() << "\n"; + llvm::errs() << " op: " << *op <<"\n"; SmallVector operandOperands; for (auto operand : operands) { if (auto broadcastInDimOp =