Package | Description |
---|---|
org.bytedeco.pytorch.cuda |
Modifier and Type | Method and Description |
---|---|
AllocatorState |
CUDAAllocator.getCheckpointState(int device,
DeviceAssertionsDataVectorCUDAKernelLaunchInfoVectorPair id) |
Modifier and Type | Method and Description |
---|---|
CheckpointDelta |
CUDAAllocator.setCheckpointPoolState(int device,
AllocatorState pps) |
Copyright © 2024. All rights reserved.