pytorch
50beab29 - [MPS] Fix the failure with ReplicatePad3D (#96988)

Commit
1 year ago
[MPS] Fix the failure with ReplicatePad3D (#96988) - Only ReflectPad needs the torch checks for input arguments and not the ReplicatePad - Added a test case - The failure was originally found in test_modules with test `test_forward_nn_ReplicationPad3d_mps_float32` Pull Request resolved: https://github.com/pytorch/pytorch/pull/96988 Approved by: https://github.com/DenisVieriu97
Author
Committer
Parents
Loading