From 166bdbff1c836b219a309bfc3293760a23a31a0e Mon Sep 17 00:00:00 2001
From: wutiancheng <wutiancheng@huawei.com>
Date: Wed, 21 Sep 2022 19:27:57 +0800
Subject: [PATCH] fix reverse sequence ops type bugs r1.9

---
 .../cpu/kernel/reverse_sequence_cpu_kernel.cc | 24 ++++++++++++++++++-
 .../cpu/kernel/reverse_sequence_cpu_kernel.h  |  9 ++++---
 2 files changed, 29 insertions(+), 4 deletions(-)

diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.cc
index 7a4494b7f93..f0ef03dbd88 100644
--- a/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.cc
+++ b/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.cc
@@ -185,6 +185,17 @@ std::vector<
      &ReverseSequenceCpuKernelMod::LaunchKernel<float, int32_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<float>},
     {KernelAttr().AddInputAttr(kNumberTypeFloat64).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeFloat64),
      &ReverseSequenceCpuKernelMod::LaunchKernel<double, int32_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<double>},
+    {KernelAttr().AddInputAttr(kNumberTypeComplex64).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeComplex64),
+     &ReverseSequenceCpuKernelMod::LaunchKernel<complex64, int32_t>,
+     &ReverseSequenceCpuKernelMod::ResizeKernel<complex64>},
+    {KernelAttr()
+       .AddInputAttr(kNumberTypeComplex128)
+       .AddInputAttr(kNumberTypeInt32)
+       .AddOutputAttr(kNumberTypeComplex128),
+     &ReverseSequenceCpuKernelMod::LaunchKernel<complex128, int32_t>,
+     &ReverseSequenceCpuKernelMod::ResizeKernel<complex128>},
+    {KernelAttr().AddInputAttr(kNumberTypeBool).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeBool),
+     &ReverseSequenceCpuKernelMod::LaunchKernel<bool, int32_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<bool>},
     {KernelAttr().AddInputAttr(kNumberTypeInt8).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeInt8),
      &ReverseSequenceCpuKernelMod::LaunchKernel<int8_t, int64_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<int8_t>},
     {KernelAttr().AddInputAttr(kNumberTypeInt16).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeInt16),
@@ -207,7 +218,18 @@ std::vector<
     {KernelAttr().AddInputAttr(kNumberTypeFloat32).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeFloat32),
      &ReverseSequenceCpuKernelMod::LaunchKernel<float, int64_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<float>},
     {KernelAttr().AddInputAttr(kNumberTypeFloat64).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeFloat64),
-     &ReverseSequenceCpuKernelMod::LaunchKernel<double, int64_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<double>}};
+     &ReverseSequenceCpuKernelMod::LaunchKernel<double, int64_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<double>},
+    {KernelAttr().AddInputAttr(kNumberTypeComplex64).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeComplex64),
+     &ReverseSequenceCpuKernelMod::LaunchKernel<complex64, int64_t>,
+     &ReverseSequenceCpuKernelMod::ResizeKernel<complex64>},
+    {KernelAttr()
+       .AddInputAttr(kNumberTypeComplex128)
+       .AddInputAttr(kNumberTypeInt64)
+       .AddOutputAttr(kNumberTypeComplex128),
+     &ReverseSequenceCpuKernelMod::LaunchKernel<complex128, int64_t>,
+     &ReverseSequenceCpuKernelMod::ResizeKernel<complex128>},
+    {KernelAttr().AddInputAttr(kNumberTypeBool).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeBool),
+     &ReverseSequenceCpuKernelMod::LaunchKernel<bool, int64_t>, &ReverseSequenceCpuKernelMod::ResizeKernel<bool>}};
 
 std::vector<KernelAttr> ReverseSequenceCpuKernelMod::GetOpSupport() {
   std::vector<KernelAttr> support_list;
diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.h b/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.h
index 8be91125b22..b72628430fd 100644
--- a/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.h
+++ b/mindspore/ccsrc/plugin/device/cpu/kernel/reverse_sequence_cpu_kernel.h
@@ -14,8 +14,8 @@
  * limitations under the License.
  */
 
-#ifndef MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_CPU_REVERSE_SEQUENCE_CPU_KERNEL_H_
-#define MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_CPU_REVERSE_SEQUENCE_CPU_KERNEL_H_
+#ifndef MINDSPORE_CCSRC_PLUGIN_DEVICE_CPU_KERNEL_REVERSE_SEQUENCE_CPU_KERNEL_H_
+#define MINDSPORE_CCSRC_PLUGIN_DEVICE_CPU_KERNEL_REVERSE_SEQUENCE_CPU_KERNEL_H_
 
 #include <vector>
 #include <map>
@@ -24,11 +24,14 @@
 #include <utility>
 #include <string>
 #include <tuple>
+#include <complex>
 #include "plugin/device/cpu/kernel/cpu_kernel.h"
 #include "plugin/factory/ms_factory.h"
 
 namespace mindspore {
 namespace kernel {
+using complex64 = std::complex<float>;
+using complex128 = std::complex<double>;
 constexpr auto kUnknown = "Unknown";
 
 class ReverseSequenceCpuKernelMod : public NativeCpuKernelMod {
@@ -84,4 +87,4 @@ class ReverseSequenceCpuKernelMod : public NativeCpuKernelMod {
 };
 }  // namespace kernel
 }  // namespace mindspore
-#endif  // MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_CPU_REVERSE_SEQUENCE_CPU_KERNEL_H_
+#endif  // MINDSPORE_CCSRC_PLUGIN_DEVICE_CPU_KERNEL_REVERSE_SEQUENCE_CPU_KERNEL_H_