V-aerus opened a new pull request, #19936:
URL: https://github.com/apache/tvm/pull/19936
This PR fixes Relax bucketize lowering to pass the correct integer output
dtype to TOPI searchsorted.
Previously, both LegalizeOps and DispatchSortScan passed the input tensor
dtype as the output dtype. For float input tensors, this caused TOPI
searchsorted to receive a float output dtype, which later failed during
binary-search lowering because bucket indices must be integer values.
This patch derives the output dtype from bucketize's out_int32 attribute:
- int32 when out_int32=True
- int64 otherwise
A numerical ExportedProgram frontend test is added to cover:
- right=False
- right=True
- out_int32=False
- out_int32=True
- float input values on bucket boundaries
Test:
python -m pytest tests/python/relax/test_frontend_from_exported_program.py
-k "bucketize" -q
--
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]
