commit 4c70ab26ef0a596d779ec38e9bc9c1bfc4373f93
author Nikita Shulga <2453524+malfet@users.noreply.github.com> Mon Mar 25 16:57:35 2024 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Mon Mar 25 16:57:35 2024 +0000
tree e19473a18a299ed303b7bdeb369868738528fdc4
parent e6a37eeb062e761f5e91afa1eee6f231f1bdc685

[MPS] Enable `index_select` for complex types (#122590)

Surprisingly, as of MacOS-14.14 MPS `gatherWithUpdatesTensor:indicesTensor:axis:batchDimensions:name:` still does not support complex types, so emulate them by using `at::view_as_real` trick

Fixes https://github.com/pytorch/pytorch/issues/122427

Pull Request resolved: https://github.com/pytorch/pytorch/pull/122590
Approved by: https://github.com/Skylion007

test/test_mps.py

diff --git a/test/test_mps.py b/test/test_mps.py
index 7e62b55..6f847e8 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -264,6 +264,7 @@
         'H',
         'hsplit',
         'imag',
+        'index_select',
         'isfinite',
         'isinf',
         'isreal',
@@ -297,6 +298,7 @@
         'randn',
         'ravel',
         'real',
+        'repeat_interleave',
         'reshape_as',
         'reshape',
         'resolve_conj',