Searched refs:setCheckpointPoolState (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/c10/cuda/ |
| D | CUDACachingAllocator.h | 288 virtual CheckpointDelta setCheckpointPoolState(
365 inline CheckpointDelta setCheckpointPoolState( in setCheckpointPoolState() function
368 return get()->setCheckpointPoolState(device, std::move(pps)); in setCheckpointPoolState()
|
| D | CUDAMallocAsyncAllocator.cpp | 658 CheckpointDelta setCheckpointPoolState( in setCheckpointPoolState() function
|
| D | CUDACachingAllocator.cpp | 1841 RestoreResult setCheckpointPoolState(PrivatePoolState& pps) { in setCheckpointPoolState() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator
3421 CheckpointDelta setCheckpointPoolState( in setCheckpointPoolState() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator
3429 auto rr = device_allocator[device]->setCheckpointPoolState(*pps); in setCheckpointPoolState()
|
| /external/pytorch/torch/csrc/cuda/ |
| D | CUDAPluggableAllocator.h | 147 c10::cuda::CUDACachingAllocator::CheckpointDelta setCheckpointPoolState(
|
| D | CUDAPluggableAllocator.cpp | 322 setCheckpointPoolState( in setCheckpointPoolState() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
|
| D | Module.cpp | 1346 auto delta = c10::cuda::CUDACachingAllocator::setCheckpointPoolState( in registerCudaPluggableAllocator()
|
| /external/pytorch/test/ |
| D | test_cuda.py | 3999 def setCheckpointPoolState( member in TestBlockStateAbsorption
4021 self.setCheckpointPoolState(device, state, [], [])
4088 self.setCheckpointPoolState(outputs[0].device.index, state, outputs2, [])
4130 self.setCheckpointPoolState(device, state, [], [])
4180 self.setCheckpointPoolState(
|