PengYoun9 commented on issue #17546: URL: https://github.com/apache/tvm/issues/17546#issuecomment-2556449649
It seems that this line of code caused it: https://github.com/apache/tvm/blob/567eeed38bdbcefb68e36328af6ab1501a81d51e/python/tvm/relay/frontend/pytorch.py#L2049 In this case, the A matrix is four-dimensional, and the B matrix is two-dimensional. It seems that broadcasting is not required when performing Matmul. Am I understanding this correctly? Similar situations might exist, and I think whether broadcasting is necessary may require stricter condition checks. @mshr-h -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
