| commit | a8acd6c4106e22ec7e6f6ab4b59016df102fc3aa | [log] [tgz] |
|---|---|---|
| author | mingfeima <[email protected]> | Tue Dec 12 12:59:47 2023 +0000 |
| committer | PyTorch MergeBot <[email protected]> | Tue Dec 12 12:59:47 2023 +0000 |
| tree | 6fbb4f8ac5d2a0433e5d89288f7b9dcd69091023 | |
| parent | 92fd3927b0797594f6419869f5060af48cca6907 [diff] [blame] |
Add Half support for AvgPool2d on CPU (#109578) Add Half support for AvgPool2d (both channels last and channels first) on CPU Pull Request resolved: https://github.com/pytorch/pytorch/pull/109578 Approved by: https://github.com/mingfeima, https://github.com/albanD
diff --git a/test/test_mps.py b/test/test_mps.py index f9f5558..6347d2e 100644 --- a/test/test_mps.py +++ b/test/test_mps.py
@@ -819,6 +819,9 @@ SKIPLIST = { 'all': None, 'any': None, + # Unsupported + # input types 'tensor<1x3x9x9xf16>' and 'tensor<1xf32>' are not broadcast compatible + 'nn.functional.avg_pool2d': [torch.float16], } def addDecorator(op, d) -> None: