Browse Source

Pre Merge pull request !1890 from 周超/c78

pull/1890/MERGE
周超 Gitee 3 years ago
parent
commit
64e44fb047
3 changed files with 4 additions and 0 deletions
  1. +1
    -0
      ge/graph/manager/graph_caching_allocator.cc
  2. +2
    -0
      ge/hybrid/common/tensor_value.h
  3. +1
    -0
      ge/hybrid/executor/hybrid_model_async_executor.cc

+ 1
- 0
ge/graph/manager/graph_caching_allocator.cc View File

@@ -163,6 +163,7 @@ Status CachingAllocator::Free(uint8_t *ptr, uint32_t device_id) {
}

std::lock_guard<std::recursive_mutex> lock(mutex_);
GELOGI("=============block ptr is %p",ptr);
auto it = allocated_blocks_.find(ptr);
if (it == allocated_blocks_.end()) {
REPORT_INNER_ERROR("E19999", "Param ptr not allocated before, device_id:%u, check invalid", device_id);


+ 2
- 0
ge/hybrid/common/tensor_value.h View File

@@ -41,8 +41,10 @@ class TensorBuffer {
~TensorBuffer();

void* Release() {
GELOGI("===========buffer is %p",buffer_);
auto ret = buffer_;
buffer_ = nullptr;
GELOGI("============ret is %p", ret);
return ret;
}



+ 1
- 0
ge/hybrid/executor/hybrid_model_async_executor.cc View File

@@ -489,6 +489,7 @@ void HybridModelAsyncExecutor::BuildDeviceTensor(TensorValue &output_tensor, GeT
GELOGD("Mem type is %d", static_cast<uint32_t>(mem_type));
auto deleter = [=](uint8_t *device_data) {
if (device_data != nullptr) {
GELOGI("Free device addr int build device tensor is %p",device_data);
if (mem_type == RDMA_HBM) {
MemManager::Instance().RdmaPoolInstance(RT_MEMORY_HBM).Free(device_data, device_id_);
} else if (mem_type == HOST_DDR) {


Loading…
Cancel
Save