Ubospica opened a new pull request, #16051:
URL: https://github.com/apache/tvm/pull/16051
This PR fixes dtype issues for
- the strides parameter of match_buffer.
- When stride is a tir var, it would become a string like `"A_s0"` in the
printed script.
- If the strides are int64 vars, when we parse the printed script, the
dtype of the strides will be set to int32 by default. That causes dtype
misalign and effects the structural_equal check.
- We will cast the string stride to the dtype of shape (which is always a
primexpr)
- ramp node. If the dtypes of base and stride of a ramp node do not match,
the dtype of stride will be casted to the dtype of base.
--
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]
