m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix memory_reuse_pass memory_size calculation error, test=develop (#19020)

Browse Source
padding_in_crf
Zeng Jinle 6 years ago committed by GitHub
parent de975be1ec
commit 2175d19993
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File

4
paddle/fluid/framework/ir/memory_optimize_pass/memory_reuse_pass.cc
Unescape View File

@ -100,8 +100,10 @@ VarDesc *MemoryReusePass::GetVarDesc(const details::VarHandle &var) const {
int64_t MemoryReusePass::GetMemorySize(const details::VarHandle &var) const { int64_t MemoryReusePass::GetMemorySize(const details::VarHandle &var) const {
auto *var_desc = GetVarDesc(var); auto *var_desc = GetVarDesc(var);
auto shapes = var_desc->GetShape(); auto shapes = var_desc->GetShape();
auto sizeof_dtype = static_cast<int64_t>(SizeOfType(var_desc->GetDataType()));
return std::accumulate(shapes.begin(), shapes.end(), static_cast<int64_t>(1), return std::accumulate(shapes.begin(), shapes.end(), static_cast<int64_t>(1),
std::multiplies<int64_t>()); std::multiplies<int64_t>()) *
sizeof_dtype;
} }
void MemoryReusePass::CollectShareTensorBufferOpHandles() const { void MemoryReusePass::CollectShareTensorBufferOpHandles() const {

Write Preview
Loading…
Cancel
Save