Add more verbose loggign for if_op shape errors

aee3164f · Ted Themistokleous · 98bf0b8b · aee3164f
Commit aee3164f authored Aug 03, 2022 by Ted Themistokleous
Hide 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};