Home
last modified time | relevance | path

Searched defs:CachingHostAllocatorImpl (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DCachingHostAllocator.h111 struct CachingHostAllocatorImpl { struct
116 virtual std::pair<void*, void*> allocate(size_t size) { in allocate()
143 virtual void free(void* ctx) { in free()
182 virtual bool record_event(void* ptr, void* ctx, S stream) { in record_event()
212 virtual void empty_cache() { in empty_cache()
236 inline size_t size_index(size_t size) { in size_index()
240 …est [[maybe_unused]], const void* src [[maybe_unused]], std::size_t count [[maybe_unused]]) const { in copy_data()
245 virtual void add_allocated_block(B* block) { in add_allocated_block()
251 virtual B* get_free_block(size_t size) { in get_free_block()
263 virtual void process_events() { in process_events()
[all …]