refine err msg of allocator, test=develop (#20879)

yaoxuefeng
Zeng Jinle 6 years ago committed by GitHub
parent ff0886a92a
commit c51722c820
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -24,13 +24,15 @@ MetadataCache::MetadataCache(bool uses_gpu) : uses_gpu_(uses_gpu) {}
MemoryBlock::Desc* MetadataCache::LoadDesc(MemoryBlock* block) {
if (uses_gpu_) {
auto existing_desc = cache_.find(block);
PADDLE_ENFORCE_EQ(existing_desc->second.CheckGuards(), true);
return &(existing_desc->second);
auto iter = cache_.find(block);
PADDLE_ENFORCE_NE(iter, cache_.end());
auto* desc = &(iter->second);
PADDLE_ENFORCE_EQ(desc->CheckGuards(), true, "Invalid CPU memory access");
return desc;
} else {
auto* desc = reinterpret_cast<MemoryBlock::Desc*>(block);
VLOG(10) << "Load MemoryBlock::Desc type=" << desc->type;
PADDLE_ENFORCE_EQ(desc->CheckGuards(), true);
PADDLE_ENFORCE_EQ(desc->CheckGuards(), true, "Invalid CPU memory access");
return reinterpret_cast<MemoryBlock::Desc*>(block);
}
}

Loading…
Cancel
Save