From 2f3f5765c2f97eb04a89a44695349a666ddabece Mon Sep 17 00:00:00 2001
From: CaoJian <caojian05>
Date: Tue, 5 Jan 2021 19:27:16 +0800
Subject: [PATCH] fix bug: Equal/NotEqual ops support int64.

---
 .../backend/kernel_compiler/cpu/arithmetic_cpu_kernel.h     | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.h
index aeee5eda18..84f162e51e 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.h
+++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.h
@@ -156,6 +156,9 @@ MS_REG_CPU_KERNEL(
 MS_REG_CPU_KERNEL(
   Equal, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeBool),
   ArithmeticCPUKernel);
+MS_REG_CPU_KERNEL(
+  Equal, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeBool),
+  ArithmeticCPUKernel);
 MS_REG_CPU_KERNEL(
   Equal, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTypeBool),
   ArithmeticCPUKernel);
@@ -186,6 +189,9 @@ MS_REG_CPU_KERNEL(
 MS_REG_CPU_KERNEL(
   NotEqual, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeBool),
   ArithmeticCPUKernel);
+MS_REG_CPU_KERNEL(
+  NotEqual, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeBool),
+  ArithmeticCPUKernel);
 MS_REG_CPU_KERNEL(
   NotEqual, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTypeBool),
   ArithmeticCPUKernel);