Package | Description |
---|---|
org.nd4j.jita.allocator.context.impl | |
org.nd4j.jita.allocator.pointers.cuda | |
org.nd4j.jita.flow | |
org.nd4j.jita.flow.impl |
Modifier and Type | Method and Description |
---|---|
protected cublasHandle_t |
BasicContextPool.createNewCublasHandle(cudaStream_t stream) |
Modifier and Type | Method and Description |
---|---|
void |
cudaEvent_t.register(cudaStream_t stream) |
Modifier and Type | Method and Description |
---|---|
void |
FlowController.commitTransfer(cudaStream_t streamUsed)
This method should be called after memcpy operations, to control their flow.
|
Modifier and Type | Method and Description |
---|---|
void |
SynchronousFlowController.commitTransfer(cudaStream_t streamUsed) |
void |
AsynchronousFlowController.commitTransfer(cudaStream_t streamUsed)
Deprecated.
|
Copyright © 2016. All Rights Reserved.