add complex64/128 ut for matrix_power

test/legacy_test/test_matrix_power_op.py

@@ -240,6 +240,82 @@ def config(self):
         self.n = -1
 
 
+class TestMatrixPowerOpCP64(TestMatrixPowerOp):
+    def config(self):
+        self.matrix_shape = [10, 10]
+        self.dtype = "complex64"
+        self.n = 2
+
+    def test_grad(self):
+        self.check_grad(["X"], "Out", max_relative_error=1e-5)
+
+
+class TestMatrixPowerOpBatchedCP64(TestMatrixPowerOpCP64):
+    def config(self):
+        self.matrix_shape = [2, 8, 4, 4]
+        self.dtype = "complex64"
+        self.n = 2
+
+
+class TestMatrixPowerOpLarge1CP64(TestMatrixPowerOpCP64):
+    def config(self):
+        self.matrix_shape = [32, 32]
+        self.dtype = "complex64"
+        self.n = 2
+
+
+class TestMatrixPowerOpLarge2CP64(TestMatrixPowerOpCP64):
+    def config(self):
+        self.matrix_shape = [10, 10]
+        self.dtype = "complex64"
+        self.n = 32
+
+
+class TestMatrixPowerOpCP64Minus(TestMatrixPowerOpCP64):
+    def config(self):
+        self.matrix_shape = [10, 10]
+        self.dtype = "complex64"
+        self.n = -1
+
+
+class TestMatrixPowerOpCP128(TestMatrixPowerOp):
+    def config(self):
+        self.matrix_shape = [10, 10]
+        self.dtype = "complex128"
+        self.n = 2
+
+    def test_grad(self):
+        self.check_grad(["X"], "Out", max_relative_error=1e-5)
+
+
+class TestMatrixPowerOpBatchedCP128(TestMatrixPowerOpCP128):
+    def config(self):
+        self.matrix_shape = [2, 8, 4, 4]
+        self.dtype = "complex128"
+        self.n = 2
+
+
+class TestMatrixPowerOpLarge1CP128(TestMatrixPowerOpCP128):
+    def config(self):
+        self.matrix_shape = [32, 32]
+        self.dtype = "complex128"
+        self.n = 2
+
+
+class TestMatrixPowerOpLarge2CP128(TestMatrixPowerOpCP128):
+    def config(self):
+        self.matrix_shape = [10, 10]
+        self.dtype = "complex128"
+        self.n = 32
+
+
+class TestMatrixPowerOpCP128Minus(TestMatrixPowerOpCP128):
+    def config(self):
+        self.matrix_shape = [10, 10]
+        self.dtype = "complex128"
+        self.n = -1
+
+
 class TestMatrixPowerAPI(unittest.TestCase):
     def setUp(self):
         np.random.seed(123)