commit 9b12a28d89084a3db70fff76c0780d1cfa65e2b0
author Nikita Shulga <nikita.shulga@gmail.com> Fri Sep 01 20:52:15 2023 -0600
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Sun Sep 10 05:39:12 2023 +0000
tree db75914791030dab189b417b6059284b68185de0
parent c7bb842d3538920fd63a7668fa9f020b1d46c224

[MPS] Implement `mul` operation for complex types (#108395)

Using existing BinaryKernel template

Add `mul` as well as `kron` and `outer` to list of MPS ops that support complex types

This should add all the missing ops mentioned in https://github.com/pytorch/pytorch/issues/105665
Pull Request resolved: https://github.com/pytorch/pytorch/pull/108395
Approved by: https://github.com/albanD
ghstack dependencies: #108393, #108394

test/test_mps.py

diff --git a/test/test_mps.py b/test/test_mps.py
index 7442f06..452bed4 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -210,6 +210,7 @@
     # Supported complex OPS
     SUPPORTED_COMPLEX_OPS = [
         '__radd__',
+        '__rmul__',
         'add',
         'atleast_1d',
         'atleast_2d',
@@ -227,11 +228,14 @@
         'isinf',
         'isreal',
         'item',
+        'kron',
         'linspace',
         'logspace',
+        'mul',
         'nn.functional.feature_alpha_dropoutwithout_train',
         'nn.functional.unfold',
         'ones',
+        'outer',
         'positive',
         'randn',
         'ravel',