torch-mlir
torch-mlir copied to clipboard
[DUPLICATE] Add onnx GridSample support for border padding mode
Duplicate of #3819, rebasing on latest changes
Adds support for padding_mode="border"
Clamps the grid coordinates between 0 and size - 1 like this when using this padding mode:
x_result = min(max(0, x), W - 1) y_result = min(max(0, y), H - 1) Added Lit tests for both TorchToLinalg and OnnxToTorch lowerings
@Ax9D Please fix the CI failures.
The new failures on nightly seems unrelated
The new failures on nightly seems unrelated
Yeah, they were because of a PyTorch version upgrade. Should be fixed now.
Is there anything else pending for this?