Home
last modified time | relevance | path

Searched refs:allocation_sizes (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/c10/mobile/
DCPUProfilingAllocator.cpp87 const std::vector<uint64_t>& allocation_sizes, in create_and_sort_mem_events() argument
90 for (uint64_t i = 0; i < allocation_sizes.size(); ++i) { in create_and_sort_mem_events()
97 events.emplace_back(i, i, allocation_sizes[i], EventType::Allocate); in create_and_sort_mem_events()
99 allocation_lifetimes[i], i, allocation_sizes[i], EventType::Free); in create_and_sort_mem_events()
111 const std::vector<uint64_t>& allocation_sizes, in formulate_greedy_allocation_plan() argument
149 allocation_sizes.size(), std::numeric_limits<uint64_t>::max()); in formulate_greedy_allocation_plan()
151 create_and_sort_mem_events(allocation_sizes, allocation_lifetimes); in formulate_greedy_allocation_plan()
241 allocation_sizes.clear(); in clear()
253 allocation_plan_->allocation_sizes.push_back(size); in record_allocation()
280 if (allocation_id_ >= allocation_plan_->allocation_sizes.size() || in validate_allocation()
[all …]
DCPUProfilingAllocator.h23 std::vector<uint64_t> allocation_sizes;