frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add avg_pool3d backward pass for MPS (#159089)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/159089
Approved by: https://github.com/malfet
This commit is contained in:
Kurt Mohler
2025-08-04 18:48:29 -05:00
committed by PyTorch MergeBot
parent 57ab39f7e4
commit b59b61a099
7 changed files with 204 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/testing/_internal/common_modules.py
View File

@ -4064,11 +4064,6 @@ module_db: list[ModuleInfo] = [
),
ModuleInfo(torch.nn.LocalResponseNorm,
module_inputs_func=module_inputs_torch_nn_LocalResponseNorm,
skips=(
# uses avg_pool3d which is not supported on MPS backend
DecorateInfo(expectedFailureMPS, 'TestModule', 'test_memory_format'),
DecorateInfo(expectedFailureMPS, 'TestModule', 'test_non_contiguous_tensors'),
DecorateInfo(expectedFailureMPS, 'TestModule', 'test_non_contiguous'),)
),
ModuleInfo(torch.nn.LayerNorm,
module_inputs_func=module_inputs_torch_nn_LayerNorm,