Java Pointer.address方法代码示例

import org.bytedeco.javacpp.Pointer; //导入方法依赖的package包/类
/**
 * Asynchronous version of memcpy
 *
 * PLEASE NOTE: This is device-dependent method, if it's not supported in your environment, blocking call will be used instead.
 *
 * @param dstBuffer
 * @param srcPointer
 * @param length
 * @param dstOffset
 */
@Override
public void memcpyAsync(DataBuffer dstBuffer, Pointer srcPointer, long length, long dstOffset) {
    AllocationPoint point = ((BaseCudaDataBuffer) dstBuffer).getAllocationPoint();
    // we update host memory regardless.
    //Pointer dP = new Pointer((point.getAllocationStatus() == AllocationStatus.DEVICE ? point.getPointers().getDevicePointer().address() : point.getPointers().getHostPointer().address()) + dstOffset);
    Pointer dP = new CudaPointer((point.getPointers().getHostPointer().address()) + dstOffset);
    //        Pointer sP = new Pointer(srcPointer.getNativePointer());
    //log.info("Location: " + point.getAllocationStatus());
    //        if (length > 4)
    //log.info("memcpyAsync:  ["+ srcPointer.getNativePointer()+"] -> ["+ dP.getNativePointer()+"], length: [" + length+ "], offset: ["+ dstOffset+"], dstBufferOffset: ["+(dstBuffer.getElementSize() * dstBuffer.offset()) + "/" + dstBuffer.offset() +"]");

    CudaContext tContext = null;

    if (dstBuffer.isConstant()) {

        org.bytedeco.javacpp.Pointer dstPointer =
                        new CudaPointer(point.getPointers().getHostPointer().address() + dstOffset, 0L);
        org.bytedeco.javacpp.Pointer srcPointerJ = new CudaPointer(srcPointer, length);

        //   log.info("JCPP Memcpy: [{}] -> [{}], length: [{}]", srcPointerJ.address(), dstPointer.address(), length);

        org.bytedeco.javacpp.Pointer.memcpy(dstPointer, srcPointerJ, length);

        point.tickHostRead();
    } else {
        //log.info("Memcpy pointers: [{}] -> [{}]", srcPointer.address(),  dP.address());

        CudaContext context = flowController.prepareAction(point);
        tContext = context;
        if (nativeOps.memcpyAsync(dP, srcPointer, length, CudaConstants.cudaMemcpyHostToHost,
                        context.getSpecialStream()) == 0)
            throw new IllegalStateException(
                            "MemcpyAsync H2H failed: [" + srcPointer.address() + "] -> [" + dP.address() + "]");

        flowController.commitTransfer(tContext.getSpecialStream());

        if (point.getAllocationStatus() == AllocationStatus.HOST)
            flowController.registerAction(context, point);
    }

    // if we're copying something into host memory, but we're on device - we need to provide exact copy to device as well
    if (point.getAllocationStatus() == AllocationStatus.DEVICE) {
        // TODO: this sounds wrong, and probably memcpy whould check initial direction, like relocate did before
        Pointer rDP = new CudaPointer(point.getPointers().getDevicePointer().address() + dstOffset);

        if (tContext == null)
            tContext = flowController.prepareAction(point);
        //log.info("MemcpyAsync to device... [{}] -> [{}]", dP.getNativePointer(), rDP.getNativePointer());

        if (nativeOps.memcpyAsync(rDP, dP, length, CudaConstants.cudaMemcpyHostToDevice,
                        tContext.getSpecialStream()) == 0)
            throw new IllegalStateException(
                            "MemcpyAsync H2D failed: [" + dP.address() + "] -> [" + rDP.address() + "]");

        flowController.commitTransfer(tContext.getSpecialStream());

        flowController.registerAction(tContext, point);


    }
    point.tickDeviceWrite();
}
 

import org.bytedeco.javacpp.Pointer; //导入方法依赖的package包/类
/**
 * This method returns
 * PLEASE NOTE: Cache options depend on specific implementations
 *
 * @param bytes
 * @param kind
 * @param initialize
 */
@Override
public Pointer allocate(long bytes, MemoryKind kind, boolean initialize) {
    Pointer ptr = NativeOpsHolder.getInstance().getDeviceNativeOps().mallocHost(bytes, 0);

    if (ptr == null || ptr.address() == 0L)
        throw new ND4JIllegalStateException("Failed to allocate [" + bytes + "] bytes");

    //log.info("Allocating {} bytes at MemoryManager", bytes);


    if (initialize)
        Pointer.memset(ptr, 0, bytes);

    return ptr;
}
 

import org.bytedeco.javacpp.Pointer; //导入方法依赖的package包/类
public LongPointerWrapper(Pointer pointer) {
    this.address = pointer.address();
    this.capacity = pointer.capacity();
    this.limit = pointer.limit();
    this.position = pointer.position();
}