[MPS] Add complex `add`/`sub` (#108394)
Using `view_as_real` and running elementwise ops in resulted tensors
Add `add` and `sub` to list of complex ops that should work on MPS
Pull Request resolved: https://github.com/pytorch/pytorch/pull/108394
Approved by: https://github.com/albanD
ghstack dependencies: #108393
diff --git a/test/test_mps.py b/test/test_mps.py
index 9d1197d..7442f06 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -209,6 +209,8 @@
def mps_ops_modifier(ops):
# Supported complex OPS
SUPPORTED_COMPLEX_OPS = [
+ '__radd__',
+ 'add',
'atleast_1d',
'atleast_2d',
'atleast_3d',
@@ -243,6 +245,7 @@
'split',
'squeeze',
'squeezemultiple',
+ 'sub',
't',
'unflatten',
'unsafe_split',
