Browse Source

modif deleter

tags/v1.3.0
zhou_chao1993 3 years ago
parent
commit
3d7d2da806
2 changed files with 6 additions and 7 deletions
  1. +5
    -6
      ge/graph/manager/host_mem_manager.cc
  2. +1
    -1
      metadef

+ 5
- 6
ge/graph/manager/host_mem_manager.cc View File

@@ -45,12 +45,11 @@ Status SharedMemAllocator::Allocate(SharedMemInfo &mem_info) {
return GE_GRAPH_MEMORY_ALLOC_FAILED;
}
mem_info.fd = output_para.fd;
mem_info.host_aligned_ptr = AlignedPtr::BuildFromAllocFunc([&output_para](std::unique_ptr<uint8_t[], deleter> &ptr) {
ptr.reset(reinterpret_cast<uint8_t *>(output_para.ptr));
},
[](uint8_t *ptr) {
ptr = nullptr;
});
mem_info.host_aligned_ptr = AlignedPtr::BuildFromAllocFunc(
[&output_para](std::unique_ptr<uint8_t[], AlignedPtr::Deleter> &ptr) {
ptr.reset(reinterpret_cast<uint8_t *>(output_para.ptr));
},
[](uint8_t *ptr) { ptr = nullptr; });
mem_info.device_address = reinterpret_cast<uint8_t *>(output_para.devPtr);
return SUCCESS;
}


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 22ab76ecd8461f679606374be4b3b6b6f7cad321
Subproject commit 1c41e02f73b6e8f95369e052ee4de285145fb34f

Loading…
Cancel
Save