| 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.