public interface FlowController
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) |
void init(Allocator allocator)
void synchronizeToHost(AllocationPoint point)
point
- void waitTillFinished(AllocationPoint point)
point
- void registerAction(CudaContext context, INDArray result, INDArray... operands)
result
- operands
- CudaContext prepareAction(INDArray result, INDArray... operands)
result
- operands
- CudaContext prepareAction(AllocationPoint result, AllocationPoint... operands)
void registerAction(CudaContext context, AllocationPoint result, AllocationPoint... operands)
void waitTillReleased(AllocationPoint point)
void commitTransfer(cudaStream_t streamUsed)
Copyright © 2016. All Rights Reserved.