as4230 opened a new pull request, #19497:
URL: https://github.com/apache/tvm/pull/19497
`topi.scatter_elements` and `topi.scatter_nd` emit bare `T.parallel` loops
in their te.extern IRBuilder bodies which trips `VerifyMemory` on CUDA targets:
RuntimeError: Memory verification failed
...
Did you forget to bind?
CPU (LLVM) is unaffected.
This fix makes the IRBuilder body in both `topi/scatter_elements.py` and
`topi/scatter.py` target-aware. When `Target.current()` is a GPU target it
emits thread bindings instead of `T.parallel`.
Fixes #19451.
--
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]
