Move the statement out of assertion.

Signed-off-by: hanhanW <[email protected]>
iree-org · Jan 13, 2025 · c2756a0 · c2756a0
1 parent 24b9679
commit c2756a0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/compiler/src/iree/compiler/Dialect/Stream/Transforms/SpecializeEncodings.cpp b/compiler/src/iree/compiler/Dialect/Stream/Transforms/SpecializeEncodings.cpp
@@ -193,9 +193,13 @@ static LogicalResult duplicateExecutablesPerAffinityVariant(
     SmallVector<Attribute> newEntryPoints;
     SmallVector<IREE::Stream::AffinityAttr> execAffinities;
     // Sanity checks. It should already meet the requirement because they are
-    // checked in step 1.
-    assert(affinityAnalysis.tryLookupExecutionAffinity(dispatchOp,
-                                                       execAffinities));
+    // checked in step 1. This can not be wrapped by an assertion because it
+    // could be dropped by compiler.
+    if (!affinityAnalysis.tryLookupExecutionAffinity(dispatchOp,
+                                                     execAffinities)) {
+      return failure();
+    }
+
     assert(execAffinities.size() == 1);
     SmallVector<Attribute> operandAttrs = resourceAffinities[dispatchOp];
     dispatchOp.forEachEntryPointAttr([&](SymbolRefAttr entryPoint) {