[Unity][CUTLASS] Require the residual input to have the same shape as input

[masahi]

In SD UNet, I met a pattern conv2d -> bias_add -> residual_add where the input shape is (2, 16, 16, 1280) and the residual input shape is (2, 1, 1, 1280). Such general broadcast add for the residual block is not supported by cutlass.

This was the cause of nan output when using the cutlass BYOC with an improved AMP pass (that allows bias add to be done in fp16, PR later).

@vinx13 @yelite @spectrometerHBH

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• cc @quic-sanirudh _{See #10317 for details}

_{Generated by tvm-bot}