public class SynchronousFlowController extends Object implements FlowController
Modifier and Type | Field and Description |
---|---|
protected Configuration |
configuration |
protected EventsProvider |
eventsProvider |
protected NativeOps |
nativeOps |
Constructor and Description |
---|
SynchronousFlowController() |
Modifier and Type | Method and Description |
---|---|
void |
commitTransfer(cudaStream_t streamUsed)
This method should be called after memcpy operations, to control their flow.
|
void |
init(Allocator allocator) |
CudaContext |
prepareAction(AllocationPoint result,
AllocationPoint... operands) |
CudaContext |
prepareAction(INDArray result,
INDArray... operands)
This method is called before operation was executed
|
CudaContext |
prepareActionAllWrite(INDArray... operands) |
protected void |
prepareDelayedMemory(DataBuffer buffer) |
protected void |
prepareDelayedMemory(INDArray array) |
void |
registerAction(CudaContext context,
AllocationPoint result,
AllocationPoint... operands) |
void |
registerAction(CudaContext context,
INDArray result,
INDArray... operands)
This method is called after operation was executed
|
void |
registerActionAllWrite(CudaContext context,
INDArray... operands) |
void |
synchronizeToHost(AllocationPoint point)
This method makes sure HOST memory contains latest data from GPU
|
void |
waitTillFinished(AllocationPoint point)
This method ensures, that all asynchronous operations on referenced AllocationPoint are finished
|
void |
waitTillReleased(AllocationPoint point) |
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getEventsProvider
protected NativeOps nativeOps
protected Configuration configuration
protected EventsProvider eventsProvider
public void init(Allocator allocator)
init
in interface FlowController
public void synchronizeToHost(AllocationPoint point)
synchronizeToHost
in interface FlowController
point
- public void waitTillFinished(AllocationPoint point)
FlowController
waitTillFinished
in interface FlowController
public CudaContext prepareActionAllWrite(INDArray... operands)
prepareActionAllWrite
in interface FlowController
public CudaContext prepareAction(INDArray result, INDArray... operands)
FlowController
prepareAction
in interface FlowController
public void waitTillReleased(AllocationPoint point)
waitTillReleased
in interface FlowController
public void registerAction(CudaContext context, AllocationPoint result, AllocationPoint... operands)
registerAction
in interface FlowController
public void registerActionAllWrite(CudaContext context, INDArray... operands)
registerActionAllWrite
in interface FlowController
public void registerAction(CudaContext context, INDArray result, INDArray... operands)
FlowController
registerAction
in interface FlowController
public CudaContext prepareAction(AllocationPoint result, AllocationPoint... operands)
prepareAction
in interface FlowController
public void commitTransfer(cudaStream_t streamUsed)
FlowController
commitTransfer
in interface FlowController
protected void prepareDelayedMemory(INDArray array)
protected void prepareDelayedMemory(DataBuffer buffer)
Copyright © 2016. All Rights Reserved.