The GitHub Actions job "Teams" on tvm.git/main has succeeded. Run started by GitHub user tlopex (triggered by tlopex).
Head commit for run: 1e08eb2fa9dcb59a4f34917a94c659d81ad36c0c / YinHanke <[email protected]> [Relax][ONNX][Torch] Add roi_align support and frontend integration (#18936) ## Summary Add Relax `roi_align` support and wire it through the ONNX and PyTorch frontends. ## Changes - add `relax.vision.roi_align`, including attrs, Python wrapper, struct info inference, and legalization - add TOPI `roi_align` compute and keep both legacy and aligned ROIAlign semantics - support ONNX `RoiAlign`, including `coordinate_transformation_mode` handling for `output_half_pixel` and `half_pixel` - support PyTorch `torchvision.ops.roi_align` in the exported-program frontend, including the `aligned` flag - add regression tests for Relax op inference, legalization, TVMScript parsing, ONNX frontend import, and PyTorch frontend import - add aligned ROIAlign test coverage to make sure sub-pixel RoIs no longer use the legacy `min=1.0` clamp ## Validation - `pytest tests/python/relax/test_op_vision.py -k roi_align` - `pytest tests/python/relax/test_tvmscript_parser_op_vision.py -k roi_align` - `pytest tests/python/relax/test_frontend_onnx.py -k roi_align` - `pytest tests/python/relax/test_frontend_from_exported_program.py -k roi_align` This PR completes the Relax/ONNX/Torch roi_align work tracked in #18928. Report URL: https://github.com/apache/tvm/actions/runs/23600466525 With regards, GitHub Actions via GitBox --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
