public class PackedContextPool extends BasicContextPool implements ContextPool
Modifier and Type | Field and Description |
---|---|
protected static int |
LANES_PER_THREAD |
contextsForDevices, cublasPool, cuPool, lock, logger, MAX_STREAMS_PER_DEVICE, nativeOps
Constructor and Description |
---|
PackedContextPool() |
Modifier and Type | Method and Description |
---|---|
CudaContext |
acquireContextForDevice(Integer deviceId) |
ContextPack |
acquireContextPackForDevice(Integer deviceId) |
containsContextForThread, createNewContext, createNewCublasHandle, createNewStream, getContextForDevice, getCuContextForDevice, getDeviceBuffers, resetPool
public CudaContext acquireContextForDevice(Integer deviceId)
acquireContextForDevice
in interface ContextPool
acquireContextForDevice
in class BasicContextPool
public ContextPack acquireContextPackForDevice(Integer deviceId)
acquireContextPackForDevice
in interface ContextPool
acquireContextPackForDevice
in class BasicContextPool
Copyright © 2016. All Rights Reserved.