add test with unknown number of iterations

EnzymeAD · Dec 21, 2024 · 10186d5 · 10186d5
1 parent b603993
commit 10186d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/enzyme/Enzyme/MLIR/Implementations/SCFAutoDiffOpInterfaceImpl.cpp b/enzyme/Enzyme/MLIR/Implementations/SCFAutoDiffOpInterfaceImpl.cpp
@@ -306,7 +306,7 @@ struct ForOpEnzymeOpsRemover
 
       auto newType =
           info.cachedType().cast<AutoDiffTypeInterface>().getShadowType(
-              numIters.value());
+              numIters.value_or(ShapedType::kDynamic));
       enzyme::InitOp newInit = ({
         OpBuilder::InsertionGuard guard(builder);
         builder.setInsertionPoint(info.initOp);