Add more verbose loggign for if_op shape errors

dd6540bd · Ted Themistokleous · Ted Themistokleous · d799e44e · dd6540bd
Commit dd6540bd authored Aug 03, 2022 by Ted Themistokleous Committed by Ted Themistokleous Oct 04, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/include/migraphx/op/if_op.hpp src/include/migraphx/op/if_op.hpp +2 -1

No files found.
--- a/src/include/migraphx/op/if_op.hpp
+++ b/src/include/migraphx/op/if_op.hpp
@@ -54,7 +54,8 @@ struct if_op
        if(not std::equal(
               out_shapes1.begin(), out_shapes1.end(), out_shapes0.begin(), out_shapes0.end()))
        {
-            MIGRAPHX_THROW("IF: output shapes of submodules must be the same.");
+            MIGRAPHX_THROW("IF:" + mods[0]->name() + " & " + mods[1]->name() +
+                           " output shapes of submodules must be the same.");
        }
        return {out_shapes0};