Fix max_pool2d decomposition for empty list and integer limits (#129106)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129106
Approved by: https://github.com/peterbell10, https://github.com/lezcano, https://github.com/malfet
ghstack dependencies: #129096, #129097
diff --git a/test/test_mps.py b/test/test_mps.py
index 77f3198..54b47d1 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -731,9 +731,6 @@
'nn.functional.interpolatearea': None,
'nn.functional.interpolatebicubic': None,
'nn.functional.interpolatetrilinear': None,
- # TODO: max_pool2d for integral types fails the numerical test
- 'nn.functional.max_pool2d': (integral_types() if product_version < 14.0 else
- [torch.int64, torch.int32, torch.int16, torch.int8]),
'nn.functional.max_unpool1dgrad': None,
'nn.functional.max_unpool2dgrad': None,
'nn.functional.max_unpool3dgrad': None,
@@ -911,6 +908,7 @@
# Error in TestConsistencyCPU.test_output_match_isin_cpu fails for integers,
# not reproducible in later OS. Added assert to op if used in < 14.0
'isin': [torch.int64, torch.int32, torch.int16, torch.uint8, torch.int8],
+ 'nn.functional.max_pool2d': [torch.uint8],
})
UNDEFINED_XFAILLIST = {
