Searched refs:change_current_allocator (Results 1 – 3 of 3) sorted by relevance
122 change_current_allocator
951 def change_current_allocator(allocator: _CUDAAllocator) -> None: function
583 torch.cuda.memory.change_current_allocator(new_alloc)598 torch.cuda.memory.change_current_allocator(new_alloc)