tlopex opened a new pull request, #19740:
URL: https://github.com/apache/tvm/pull/19740
This pr updates the WebGPU multi-warp allreduce test to check the generated
`tirx.volatile` allocation annotation structurally instead of matching the
exact TVMScript printer output.
The test is intended to verify that `LowerThreadAllreduce` marks the
generated shared allocation as volatile. It previously checked for the exact
string:
```python
"tirx.volatile": T.bool(True)
```
However, the current printer emits the same annotation as:
```python
annotations={"tirx.volatile": True}
```
The transform behavior is unchanged; only the printer spelling differs.
This patch walks the generated TIRX body and checks for an `AllocBuffer` with
`tirx.volatile=True`, which matches the actual semantic requirement of the test
without depending on bool literal formatting.
--
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]
