cuda.js/lib/memory.d.ts

CUDA bindings for Node.js

export declare class GpuBuffer {
    private readonly _handle;
    readonly elementType: string | undefined;
    constructor(size: number);
    constructor(bytesOrHandle: any, isHandle?: boolean);
    get ptr(): bigint;
    get size(): number;
    free(): void;
    upload(data: ArrayBufferView): void;
    download(): Uint8Array;
    memset(value: number, size?: number): void;
    copy(src: GpuBuffer, size?: number): void;
    get _isGpuBuffer(): boolean;
    static fromHandle(handle: any): GpuBuffer;
}
export declare class Memory {
    static malloc(size: number): GpuBuffer;
    static free(buffer: GpuBuffer): void;
    static toDevice(data: ArrayBufferView): GpuBuffer;
    static fromDevice(buffer: GpuBuffer, dtype?: 'uint8'): Uint8Array;
    static memset(buffer: GpuBuffer, value: number): void;
    static copy(dst: GpuBuffer, src: GpuBuffer, size?: number): void;
}
//# sourceMappingURL=memory.d.ts.map