[MPS] Lift MSL version to 3.0+ and use relevant helpers

[qqaatw]

Summary:

1. Remove the custom atomic add function and use the one provided by MSL 3.0+ instead.
2. Use MetalShaderLibrary class.

cc @malfet @kulinseth

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 2 New Failures, 16 Unrelated Failures

As of commit 21e09dc with merge base 4249b61 ():

NEW FAILURES - The following jobs have failed:

• Docs / build / linux-job (gh)
  RuntimeError: Command docker exec -t e1170bdaeb595e032391d264d8d81aebaacbe3049a9bf3f3d6c5dbcc0654b85e /exec failed with exit code 2
• Lint / c-source / linux-job (gh)
  RuntimeError: Command docker exec -t ef860c3c4cf75d3eecbe39dee67e5d7040000a96b1ea775daf48a78ecf440992 /exec failed with exit code 1

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Build Aarch64 Linux Wheels / pytorch/vision / upload / wheel-py3_9-cpu-aarch64 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cpu_aarch64
• Build Linux Conda / pytorch/vision / upload / conda-py3_9-cuda11_8 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cu118_
• Build Linux Conda / pytorch/vision / upload / conda-py3_9-cuda12_1 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cu121_
• Build Linux Conda / pytorch/vision / upload / conda-py3_9-cuda12_4 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cu124_
• Build Linux Wheels / pytorch/vision / upload / manywheel-py3_9-cpu (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cpu_x86_64
• Build Linux Wheels / pytorch/vision / upload / manywheel-py3_9-cuda11_8 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cu118_x86_64
• Build Linux Wheels / pytorch/vision / upload / manywheel-py3_9-cuda12_1 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cu121_x86_64
• Build Linux Wheels / pytorch/vision / upload / manywheel-py3_9-cuda12_4 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_cu124_x86_64
• Build Linux Wheels / pytorch/vision / upload / manywheel-py3_9-rocm6_1 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_rocm6.1_x86_64
• Build Linux Wheels / pytorch/vision / upload / manywheel-py3_9-rocm6_2 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_vision__3.9_rocm6.2_x86_64
• Tests / unittests-linux (3.10, linux.12xlarge, cpu) / linux-job (gh) (trunk failure)
  test/test_ops.py::TestDeformConv::test_backward[0-True-cpu]
• Tests / unittests-linux (3.11, linux.12xlarge, cpu) / linux-job (gh) (trunk failure)
  test/test_ops.py::TestDeformConv::test_backward[0-True-cpu]
• Tests / unittests-linux (3.12, linux.12xlarge, cpu) / linux-job (gh) (trunk failure)
  test/test_ops.py::TestDeformConv::test_backward[0-True-cpu]
• Tests / unittests-linux (3.9, linux.12xlarge, cpu) / linux-job (gh) (trunk failure)
  test/test_ops.py::TestDeformConv::test_backward[0-True-cpu]
• Tests / unittests-linux (3.9, linux.g5.4xlarge.nvidia.gpu, cuda, 11.8) / linux-job (gh) (trunk failure)
  AttributeError: '_OpNamespace' 'image' object has no attribute '_jpeg_version'
• Tests / unittests-windows (3.9, windows.g5.4xlarge.nvidia.gpu, cuda, 11.8) / windows-job (gh) (trunk failure)
  AttributeError: '_OpNamespace' 'image' object has no attribute '_jpeg_version'

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

[malfet]

Thank you for the BE change

[github-actions]

Hey @malfet!

You merged this PR, but no labels were added.
The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py