Examples with AllocationPoint org.nd4j.jita.allocator.impl.AllocationPoint used on opensource projects

Example 66 with AllocationPoint

use of org.nd4j.jita.allocator.impl.AllocationPoint in project nd4j by deeplearning4j.

the class AsynchronousFlowController method synchronizeReadLanes.

protected void synchronizeReadLanes(AllocationPoint point) {
    cudaEvent_t event;
    int cnt = 0;
    while ((event = point.getReadLane().poll()) != null) {
        event.synchronize();
        event.destroy();
        cnt++;
    }
// log.info("Events synchronized: [{}]", cnt);
}

Example 67 with AllocationPoint

use of org.nd4j.jita.allocator.impl.AllocationPoint in project nd4j by deeplearning4j.

the class AsynchronousFlowController method synchronizeReadLanes.

protected void synchronizeReadLanes(INDArray array) {
    if (array == null)
        return;
    AllocationPoint point = allocator.getAllocationPoint(array);
    synchronizeReadLanes(point);
}