quic-sanirudh commented on PR #16795: URL: https://github.com/apache/tvm/pull/16795#issuecomment-2021948537
> Yeah I think fixing the dtype is a good idea, it would hopefully avoid this kind of problems in the future as well. Out of interest, what were the mismatching dtypes of the two compared `IntImmNode`s that you observed @quic-sanirudh? Thanks @ekalda. I'll update the PR to fix the dtypes in RampNode (and perhaps the broadcast node as well). The dtypes in my case were `int32` and `int64`. The expression I saw was something like this (slightly simpler version) `T.Broadcast(c, 128) + T.Ramp(T.int64(0), T.int64(1), T.int64(128))` The RampNode seems to get the int64 lanes because the all the iterators in our case is by default int64, but the broadcast seems to be inserted during the [evaluation of AddNode in op.cc here](https://github.com/apache/tvm/blob/d43e1ab71d5d9e16bbc962d4d7952dcc7a1cdbca/src/tir/op/op.cc#L126-L139) -- 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]
