Method read_tensor

cake-core/src/utils/tensor_storage.rs:471–503

(&self, name: &str)

impl TensorStorageProvider for SafetensorsStorage {
    fn read_tensor(&self, name: &str) -> Result<TensorData> {
        let meta = self
            .index
            .get(name)
            .ok_or_else(|| anyhow::anyhow!("tensor '{}' not found in storage", name))?;

        let shard = self
            .shards
            .get(meta.shard_idx as usize)
            .ok_or_else(|| anyhow::anyhow!("shard index {} out of range", meta.shard_idx))?;

        let size = meta.byte_size as usize;

        // mmap: zero-syscall read — just memcpy from mapped memory
        #[cfg(unix)]
        let buf = shard.read_bytes(meta.abs_offset, size);

        #[cfg(not(unix))]
        let buf = {
            use std::io::{Read, Seek, SeekFrom};
            let mut f = shard.file.try_clone()?;
            let mut buf = vec![0u8; size];
            f.seek(SeekFrom::Start(meta.abs_offset))?;
            f.read_exact(&mut buf)?;
            buf
        };

        Ok(TensorData {
            bytes: buf,
            dtype: meta.dtype,
            shape: meta.shape.to_vec(),
        })
    }

    fn read_tensors(&self, names: &[&str]) -> Result<Vec<TensorData>> {
        if names.is_empty() {