commit 4d20b82812951fd971f930f43167c11e46da1c25
author Ben Vanik <ben.vanik@gmail.com> Thu Oct 24 08:36:20 2024 -0700
committer GitHub <noreply@github.com> Thu Oct 24 08:36:20 2024 -0700
tree 9485395ae7def196d843711d7e5060ea8cf83b6c
parent 9f5610d170c89df0e1c54a4a0b1c4162ee8bc565

Emit an error when affinity analysis fails. (#18883)

Includes the flag I'd tell anyone to use if they filed a bug.

Fixes #18878.

compiler/src/iree/compiler/Dialect/Stream/Transforms/ConvertToStream.cpp

diff --git a/compiler/src/iree/compiler/Dialect/Stream/Transforms/ConvertToStream.cpp b/compiler/src/iree/compiler/Dialect/Stream/Transforms/ConvertToStream.cpp
index 91f5c0f..92a3457 100644
--- a/compiler/src/iree/compiler/Dialect/Stream/Transforms/ConvertToStream.cpp
+++ b/compiler/src/iree/compiler/Dialect/Stream/Transforms/ConvertToStream.cpp
@@ -224,6 +224,11 @@
     // for all SSA values we'll use during conversion are available.
     AffinityAnalysis affinityAnalysis(getOperation());
     if (failed(affinityAnalysis.run())) {
+      getOperation().emitError()
+          << "affinity analysis failed to converge (input program may have "
+             "invalid affinities assigned); use"
+             "`--iree-stream-annotate-input-affinities` to help identify the "
+             "invalid affinities";
       return signalPassFailure();
     }