Fix for MPS regression in #122016 and #123178

[pytorchbot]

Fixes #122016 and #123178. This regression is related to an OS side change that requires a slight adjustment from us on PyTorch side to restore the previous behavior. Additionally we cleared out pre-MacOS13 related workarounds.

Before the fix on MacOS 14.4:

python -c "import torch;x=torch.zeros(3, device='mps');x[1] = 1; x[2] = 3; print(x)"
tensor([0., 3., 3.], device='mps:0')

After the fix:

python -c "import torch;x=torch.zeros(3, device='mps');x[1] = 1; x[2] = 3; print(x)"
tensor([0., 1., 3.], device='mps:0')

This also fixes complex number initialization and as such makes nn.functional.rms_norm pass on MacOS-14+

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/123385

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

Note: Links to docs will display an error until the docs builds have been completed.

❌ 24 New Failures, 2 Unrelated Failures

As of commit 7426754 with merge base 86a2d67 ():

NEW FAILURES - The following jobs have failed:

• Mac MPS / macos-12-py3-arm64-mps / test (mps, 1, 1, macos-m1-stable) (gh)
• pull / linux-docs (gh)
• pull / linux-focal-cpu-py3.10-gcc9-bazel-test / build-and-test (default, 1, 1, linux.4xlarge) (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-cuda11.8-py3.10-gcc9 (gh)
• pull / linux-focal-cuda11.8-py3.10-gcc9 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-cuda12.1-py3.10-gcc9-bazel-test / build-and-test (default, 1, 1, linux.4xlarge.nvidia.gpu) (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-cuda12.1-py3.10-gcc9-sm86 (gh)
• pull / linux-focal-cuda12.1-py3.10-gcc9-sm86 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-py3_8-clang9-xla (gh)
• pull / linux-focal-py3_8-clang9-xla / build (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-py3-clang9-android-ndk-r21e-gradle-custom-build-single-full-jit / build-and-test (default, 1, 1, linux.2xlarge) (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-py3.11-clang10 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-py3.12-clang10 (gh)
• pull / linux-focal-py3.12-clang10 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-py3.8-clang10 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-py3.8-clang10-onnx (gh)
• pull / linux-focal-py3.8-clang10-onnx / build (gh)
  ##[error]The operation was canceled.
• pull / linux-focal-rocm6.0-py3.8 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-jammy-cuda11.8-cudnn8-py3.8-clang12 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-jammy-py3-clang12-executorch / build (gh)
  ##[error]The operation was canceled.
• pull / linux-jammy-py3.10-clang15-asan / build (gh)
  ##[error]The operation was canceled.
• pull / linux-jammy-py3.8-gcc11 / build (gh)
  ##[error]The operation was canceled.
• pull / linux-jammy-py3.8-gcc11-no-ops / build (gh)
  ##[error]The operation was canceled.
• pull / linux-jammy-py3.8-gcc11-pch / build (gh)
  ##[error]The operation was canceled.

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

• Mac MPS / macos-12-py3-arm64-mps / test (mps, 1, 1, macos-m2-14) (gh)
  ##[error]The operation was canceled.
• pull / win-vs2019-cpu-py3 / build (gh)
  ##[error]The operation was canceled.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[malfet]

@pytorchbot rebase -b release/2.3

[pytorchmergebot]

@pytorchbot started a rebase job onto refs/remotes/origin/release/2.3. Check the current status here

[pytorchmergebot]

Successfully rebased cherry-pick-123234-by-pytorch_bot_bot_ onto refs/remotes/origin/release/2.3, please pull locally before adding more changes (for example, via git checkout cherry-pick-123234-by-pytorch_bot_bot_ && git pull --rebase)