Searched refs:ptr_to_block_ (Results 1 – 3 of 3) sorted by relevance
199 auto it = ptr_to_block_.find(ptr); in record_event()200 if (it != ptr_to_block_.end()) { in record_event()229 ptr_to_block_.erase(block->ptr_); in empty_cache()248 ptr_to_block_.insert({block->ptr_, block}); in add_allocated_block()341 ska::flat_hash_map<void*, B*> ptr_to_block_; member
538 ptr_to_block_.emplace(ptr, std::move(block)); in alloc()564 ptr_to_block_.erase(ptr); in free()807 auto it = ptr_to_block_.find(ptr); in find_block()808 if (it == ptr_to_block_.end()) { in find_block()
111 std::unordered_map<void*, c10::intrusive_ptr<Block>> ptr_to_block_; member in c10d::symmetric_memory::CUDASymmetricMemoryAllocator