tlopex opened a new pull request, #20038:
URL: https://github.com/apache/tvm/pull/20038
This PR improves the precision-loss warning emitted while building a
`BufferStore`.
Before this change, the warning only displayed the LHS and RHS dtypes,
making it difficult to identify the failing kernel when many kernels were built
together:
```text
Warning: Casting in BufferStore may lose precision:
LHS is `T.int32`, RHS is `T.float32`, indexing lanes: 1
```
After this change, the warning also displays the current kernel name and the
complete `BufferStore` node:
```text
Warning: Casting in BufferStore may lose precision:
LHS is `T.int32`, RHS is `T.float32`, indexing lanes: 1,
kernel: `warning_kernel`
BufferStore:
A = T.Buffer((4,), "int32")
B = T.Buffer((4,))
i = T.int32()
A[i] = T.Cast("int32", B[i] + T.float32(0.5))
```
--
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]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
