public class SynchronousFlowController extends Object implements FlowController
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
|
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 |
synchronizeToHost(AllocationPoint point)
This method ensures, that all asynchronous operations on referenced AllocationPoint are finished, and host memory state is up-to-date
|
void |
waitTillFinished(AllocationPoint point)
This method ensures, that all asynchronous operations on referenced AllocationPoint are finished
|
void |
waitTillReleased(AllocationPoint point) |
protected NativeOps nativeOps
public void init(Allocator allocator)
init
in interface FlowController
public void synchronizeToHost(AllocationPoint point)
FlowController
synchronizeToHost
in interface FlowController
public void waitTillFinished(AllocationPoint point)
FlowController
waitTillFinished
in interface FlowController
public void registerAction(CudaContext context, INDArray result, INDArray... operands)
FlowController
registerAction
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 CudaContext prepareAction(AllocationPoint result, AllocationPoint... operands)
prepareAction
in interface FlowController
public void commitTransfer(cudaStream_t streamUsed)
FlowController
commitTransfer
in interface FlowController
Copyright © 2016. All Rights Reserved.