Skip to content

Commit 3665358

Browse files
authored
Merge pull request #10475 from gmcather/develop
add prependAllocatedop
2 parents cc75e84 + 9ca8124 commit 3665358

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

paddle/fluid/framework/block_desc.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,11 @@ OpDesc *BlockDesc::PrependOp() {
134134
return ops_.front().get();
135135
}
136136

137+
void BlockDesc::PrependAllocatedOp(std::unique_ptr<OpDesc> &&op_desc) {
138+
need_update_ = true;
139+
ops_.emplace_front(std::move(op_desc));
140+
}
141+
137142
OpDesc *BlockDesc::InsertOp(size_t index) {
138143
need_update_ = true;
139144
auto it = ops_.begin() + index;

paddle/fluid/framework/block_desc.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,8 @@ class BlockDesc {
8888

8989
OpDesc *PrependOp();
9090

91+
void PrependAllocatedOp(std::unique_ptr<OpDesc> &&op_desc);
92+
9193
OpDesc *InsertOp(size_t index);
9294

9395
/*

0 commit comments

Comments
 (0)