MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / DeallocateRaw

Method DeallocateRaw

tensorflow/core/common_runtime/gpu/gpu_vmem_allocator.cc:71–80  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

69}
70
71void GPUVMemAllocator::DeallocateRaw(void* ptr) {
72 mutex_lock l(lock_);
73 if (device_ptrs_.count(ptr) > 0) {
74 device_allocator_->DeallocateRaw(ptr);
75 device_ptrs_.erase(ptr);
76 return;
77 } else {
78 host_allocator_->DeallocateRaw(ptr);
79 }
80}
81
82size_t GPUVMemAllocator::RequestedSize(const void* ptr) const {
83 mutex_lock l(lock_);

Callers 13

SetProtoFromGPUMethod · 0.45
~BaseGPUDeviceMethod · 0.45
TESTFunction · 0.45
BM_AllocationFunction · 0.45
BM_AllocationThreadedFunction · 0.45
BM_AllocationDelayedFunction · 0.45
TestBinDebugInfoMethod · 0.45
TEST_FFunction · 0.45
TESTFunction · 0.45
TESTFunction · 0.45
TESTFunction · 0.45

Calls 2

countMethod · 0.45
eraseMethod · 0.45

Tested by 11

TESTFunction · 0.36
BM_AllocationFunction · 0.36
BM_AllocationThreadedFunction · 0.36
BM_AllocationDelayedFunction · 0.36
TestBinDebugInfoMethod · 0.36
TEST_FFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36