This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
hummingbird
/
graphengine
Not watched
Unwatch
Watch all
Watch but not notify
2
Star
0
Fork
0
Code
Releases
13
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
Fix ut.
tags/v1.5.1
zhaozhixuan
4 years ago
parent
813f2fe4a2
commit
4f3f54d564
2 changed files
with
3 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
ge/single_op/task/op_task.h
+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_; //
I
nclude const input
size_t input_num_; //
i
nclude 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;
Write
Preview
Loading…
Cancel
Save