diff --git a/backends/arm/test/ops/test_grid_sampler.py b/backends/arm/test/ops/test_grid_sampler.py index c5a1f3560bd..4c8ab0de8ce 100644 --- a/backends/arm/test/ops/test_grid_sampler.py +++ b/backends/arm/test/ops/test_grid_sampler.py @@ -60,3 +60,18 @@ def test_grid_sampler_vgf_no_quant(test_data): run_on_vulkan_runtime=False, ) pipeline.run() + + +@common.parametrize("test_data", test_data_suite, xfails=xfails, strict=False) +@common.SkipIfNoModelConverter +def test_grid_sampler_vgf_quant(test_data): + test_data = test_data() + pipeline = VgfPipeline[input_t]( + GridSampler2d(), + test_data, + aten_op, + exir_op, + quantize=True, + run_on_vulkan_runtime=False, + ) + pipeline.run() diff --git a/docs/source/backends/arm-vgf/VGF_op_support.md b/docs/source/backends/arm-vgf/VGF_op_support.md index fc77f589bbc..7ae696a7642 100644 --- a/docs/source/backends/arm-vgf/VGF_op_support.md +++ b/docs/source/backends/arm-vgf/VGF_op_support.md @@ -63,8 +63,8 @@ Total supported PyTorch APIs: **154**. | `torch.full_like` | FP, INT | `FP32`, `INT8` | 8x8 | | `torch.gather` | FP, INT | `FP32`, `FP16`, `BF16`, `INT8`, `BOOL` | 8x8 | | `torch.ge` / `>=` | FP, INT | `FP32`, `INT8`, `INT16` | 8x8, 16x8 | -| `torch.grid_sampler` | FP | `FP32` | - | -| `torch.grid_sampler_2d` | FP | `FP32` | - | +| `torch.grid_sampler` | FP, INT | `FP32`, `INT8` | 8x8 | +| `torch.grid_sampler_2d` | FP, INT | `FP32`, `INT8` | 8x8 | | `torch.group_norm` | FP | `FP32` | - | | `torch.gt` / `>` | FP, INT | `FP32`, `INT8`, `INT16` | 8x8, 16x8 | | `torch.index_put_` | INT | `INT8` | 8x8 |