Thanks for doing this!!
It definitely helps learning more about the issue. Yet it is still unclear to me how that happened. If possible, could you share the relevant graph composition? Ideally one that can reproduce this issue, e.g. a subgraph that errors the same way when called backward on it.