Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp200 std::shared_ptr<GatheredContext> context_when_segment_allocated; member
1924 head_block->context_when_segment_allocated; in snapshot()
2274 if (!to_map->prev && !to_map->context_when_segment_allocated) { in map_block()
2275 to_map->context_when_segment_allocated = ctx; in map_block()
2406 dst->context_when_segment_allocated = in try_merge_blocks()
2407 std::move(src->context_when_segment_allocated); in try_merge_blocks()
2719 p.block->context_when_segment_allocated = ctx; in alloc_block()
2828 context ? context : block->context_when_segment_allocated); in release_block()
2926 context ? context : block->context_when_segment_allocated); in unmap_block()