MPS: edsr inference error

[guangy10]

🐛 Describe the bug

edsr+mps has been failing consistently: https://github.com/pytorch/executorch/actions/workflows/apple-perf.yml?query=event%3Aschedule

We will disable this model from continuous benchmarking and track the bug/fix here.

The stacktrace:

  Test Case '-[GenericTests test_forward_edsr_mps_float16_pte_iOS_17_2_1_iPhone15_4]' started.
  2025-01-23 00:51:29.808250-0500 Benchmark[472:19024] Invalid group id in source layers: test_layer
  2025-01-23 00:51:29.808899-0500 Benchmark[472:19024] Validation failure: Input N1D1C256H224W224 and output N1D64C2H224W224 tensors must have the same number of elements
  loc("view_copy"("(mpsFileLoc): /Library/Caches/com.apple.xbs/Sources/MetalPerformanceShadersGraph/mpsgraph/MetalPerformanceShadersGraph/Core/Files/MPSGraphUtilities.mm":303:0)): error: 'anec.reshape' op Invalid configuration for the following reasons: Input N1D1C256H224W224 and output N1D64C2H224W224 tensors must have the same number of elements
  libc++abi: terminating due to uncaught exception of type std::bad_optional_access: bad_optional_access

Job link: https://github.com/pytorch/executorch/actions/runs/12920386457/job/36038662255#step:14:1918

Versions

trunk

[guangy10]

cc: @kimishpatel @cbilgin @cccclai Can you tag the PoC from MPS to take a look?