update tests for grad_req change

src/ngraph/ngraph_nnvm_utils.h

@@ -19,6 +19,7 @@
 #include <functional>
 #include <vector>
 
+#include <mxnet/op_attr_types.h>
 #include "ngraph_sgcompiler_utils.h"
 
 namespace ngraph_bridge {

tests/cpp/ngraph/test_ngraph_imperative.cc

@@ -82,7 +82,7 @@ TEST_F(NGRAPH_IMPERATIVE, INVOKE_OP) {
   EXPECT_TRUE(op_ng);
   EXPECT_TRUE(test.op_ngraph_->ngraph_forward);
   EXPECT_EQ(vec3, std::vector<float>({0, 0}));
-  compute_forward(opctx, op_ng, inputs, outputs);
+  compute_forward(opctx, op_ng, inputs, req, outputs);
   EXPECT_EQ(vec3, std::vector<float>({2, 6}));
 }
 

tests/cpp/ngraph/test_ngraph_imperative.h

@@ -20,6 +20,7 @@
 #include "../../src/ngraph/ngraph_imperative.h"
 #include "../../src/ngraph/ngraph_nnvm_utils.h"
 #include "test_util.h"
+
 namespace ngraph_bridge {
 
 class NGRAPH_IMPERATIVE : public ::testing::Test {
@@ -39,6 +40,7 @@ class NGRAPH_IMPERATIVE : public ::testing::Test {
   std::vector<float> vec3{0, 0};
   std::vector<mxnet::TBlob> inputs;
   std::vector<mxnet::TBlob> outputs;
+  std::vector<mxnet::OpReqType> req{mxnet::kWriteTo};
 };
 
 class testImperative : public NGImperative {

tests/cpp/ngraph/test_ngraph_utils.cc

@@ -115,17 +115,20 @@ TEST(NGRAPH_NNVM, copy_TBlobs) {
   /*                     placeholders[1]) */
   /*                     ->get_vector<float>()); */
   std::vector<float> vec3{0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
-  std::vector<float> vec4{0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
+  std::vector<float> vec4{1, 1, 1, 1, 1, 1, 1, 1, 1, 1};
   mxnet::TBlob TBlob3(vec3.data(), shape, 0);
   mxnet::TBlob TBlob4(vec4.data(), shape, 0);
   std::vector<mxnet::TBlob> outblobs;
   outblobs.push_back(TBlob3);
   outblobs.push_back(TBlob4);
 
-  result_to_TBlob(placeholders[0], outblobs, 0);
-  result_to_TBlob(placeholders[1], outblobs, 1);
+  // test 1: kWriteTo - vec3 = vec1
+  // test 2: kAddTo - vec4 += vec2
+  std::vector<mxnet::OpReqType> req{mxnet::kWriteTo, mxnet::kAddTo};
+  result_to_TBlob(placeholders, req, outblobs);
   EXPECT_EQ(vec1, vec3);
-  EXPECT_EQ(vec2, vec4);
+  std::vector<float> vec4_plus_vec2{12, 13, 14, 15, 16, 17, 18, 19, 20, 11};
+  EXPECT_EQ(vec4_plus_vec2, vec4);
 }
 
 }  // namespace ngraph_bridge