skainet-lang-core/sk.ainet.lang.tensor.data/MmapTensorSource

MmapTensorSource

class MmapTensorSource(mappedBuffer: MappedByteBuffer) : AutoCloseable(source)

A handle to a memory-mapped file region that can provide tensor views.

This class manages the lifecycle of a memory-mapped file and provides methods to create tensor views into specific regions of the mapped data.

Usage:

val mmap = MmapTensorSource.fromFile(file)
val tensor1 = mmap.tensorAt<FP32>(offset1, shape1)
val tensor2 = mmap.tensorAt<FP32>(offset2, shape2)
// tensors share the same underlying memory-mapped region
mmap.close() // release when done

Parameters

mappedBuffer

the memory-mapped buffer for the entire file or region

Constructors

MmapTensorSource

constructor(mappedBuffer: MappedByteBuffer)

Types

Companion

object Companion

Functions

open override fun close()

floatTensorAt

fun <T : DType> floatTensorAt(byteOffset: Long, shape: Shape): MmapFloatTensorData<T>

Creates a tensor data view at the specified byte offset.

isLoaded

fun isLoaded(): Boolean

Check if the mapped memory is resident in physical memory.

load

fun load(): MmapTensorSource

Force the mapped memory to be loaded into physical memory. This can improve first-access performance but uses more memory.