CPU: Implement op::Equal

src/ngraph/runtime/cpu/emitter.cpp

@@ -252,3 +252,22 @@ void Emitter::EMITTER_DECL(EmitDivide)
                    EIGEN_VECTOR_FORMAT(inputs[1].get_layout<DenseTensorViewLayout>()->get_size()) ");\n"
           "    }\n";
 }
+
+void Emitter::EMITTER_DECL(EmitEqual)
+{
+    const element::Type& et = (dynamic_pointer_cast<const TensorViewType>(
+                                       n->get_arguments().at(0)->get_value_type()))
+                                      ->get_element_type();
+
+    TU += "    {\n"
+          "        auto arg0 = call_frame->get_tensor_view_data<" + element_type_names[TI(et)] + ">(" + to_string(inputs[0].get_index()) + ");\n"
+          "        auto arg1 = call_frame->get_tensor_view_data<" + element_type_names[TI(et)] + ">(" + to_string(inputs[1].get_index()) + ");\n"
+          "        auto out  = call_frame->get_tensor_view_data<Bool>(" + to_string(outputs[0].get_index()) + ");\n"
+          "        EigenArray1d<Bool>(out, "
+                   EIGEN_VECTOR_FORMAT(outputs[0].get_layout<DenseTensorViewLayout>()->get_size()) ") =\n"
+          "        (EigenArray1d<" + element_type_names[TI(et)] + ">(arg0, "
+                   EIGEN_VECTOR_FORMAT(inputs[0].get_layout<DenseTensorViewLayout>()->get_size()) ") ==\n"
+          "        EigenArray1d<" + element_type_names[TI(et)] + ">(arg1, "
+                   EIGEN_VECTOR_FORMAT(inputs[1].get_layout<DenseTensorViewLayout>()->get_size()) ")).template cast<char>();\n"
+          "    }\n";
+}

src/ngraph/runtime/cpu/emitter.hpp

@@ -52,6 +52,7 @@ namespace ngraph
                 void EMITTER_DECL(EmitAbs);
                 void EMITTER_DECL(EmitConcat);
                 void EMITTER_DECL(EmitDivide);
+                void EMITTER_DECL(EmitEqual);
 
             };
         }

src/ngraph/runtime/cpu/external_function.cpp

@@ -78,7 +78,8 @@ static const OpMap dispatcher{{TI(ngraph::op::Add), &Emitter::EmitAdd},
                               {TI(ngraph::op::Tuple), &Emitter::EmitTuple},
                               {TI(ngraph::op::Abs), &Emitter::EmitAbs},
                               {TI(ngraph::op::Concat), &Emitter::EmitConcat},
-                              {TI(ngraph::op::Divide), &Emitter::EmitDivide}
+                              {TI(ngraph::op::Divide), &Emitter::EmitDivide},
+                              {TI(ngraph::op::Equal), &Emitter::EmitEqual}
                              };
 
 #undef TI

test/cpu.cpp

@@ -418,8 +418,7 @@ TEST(cpu, divide)
     ASSERT_EQ((vector<float>{2, 2, 2, 2}), result->get_vector());
 }
 
-/*
-TEST(execute, equal)
+TEST(cpu, equal)
 {
     auto shape = Shape{2, 2, 2};
     auto A = make_shared<op::Parameter>(element::Float32::element_type(), shape);
@@ -427,7 +426,7 @@ TEST(execute, equal)
     auto rt = make_shared<TensorViewType>(element::Bool::element_type(), shape);
     auto f = make_shared<Function>(make_shared<op::Equal>(A, B), rt, op::Parameters{A, B});
 
-    auto manager = runtime::Manager::get("NGVM");
+    auto manager = runtime::Manager::get("CPU");
     auto external = manager->compile(f);
     auto backend = manager->allocate_backend();
     auto cf = backend->make_call_frame(external);
@@ -443,6 +442,7 @@ TEST(execute, equal)
     ASSERT_EQ((vector<char>{1, 1, 0, 0, 0, 1, 1, 0}), result->get_vector());
 }
 
+/*
 TEST(execute, dot_0_0)
 {
     auto shape = Shape{0};