Browse Source

Fix ut.

tags/v1.5.1
zhaozhixuan 4 years ago
parent
commit
4f3f54d564
2 changed files with 3 additions and 1 deletions
  1. +1
    -1
      ge/single_op/task/op_task.h
  2. +2
    -0
      tests/ut/ge/single_op/single_op_task_unittest.cc

+ 1
- 1
ge/single_op/task/op_task.h View File

@@ -124,7 +124,7 @@ class TbeOpTask : public OpTask {
std::string original_kernel_key_;
std::string node_info_;
std::vector<size_t> arg_index_; // data index in args
size_t input_num_; // Include const input
size_t input_num_; // include const input
size_t output_num_;
};



+ 2
- 0
tests/ut/ge/single_op/single_op_task_unittest.cc View File

@@ -134,6 +134,8 @@ TEST_F(UtestSingleOpTask, test_update_ioaddr) {
task.arg_size_ = sizeof(void *) * 4;
task.args_.reset(new (std::nothrow) uint8_t[task.arg_size_]);
task.arg_index_ = {0};
task.input_num_ = 2;
task.output_num_ = 1;

vector<void *> args;
vector<DataBuffer> inputs;


Loading…
Cancel
Save