c – 使用HDF5 H5Dread过多的内存

2023年3月17日 494次阅读

由于使用了太多内存,我们的程序崩溃了.这是HDF5 1.8.9.

大多数情况下,我们没有问题,但有时文件较大,会发生以下情况：

在这个例子中,我有一个325MB的HDF5文件,它可以使用2GB的内存来读取它的一些值(HDF5文件中数据点的“时间步长”,只有400,001个双精度值).看来我们使用H5Dread方法导致了这个问题.我们在这里做错了什么想法？

导致问题的方法如下所示：

std::vector<double> Hdf5DataReader::GetUnlimitedDimensionValues() 
{ 
    // Define hyperslab in the dataset 
    hid_t time_dataspace = H5Dget_space(mUnlimitedDatasetId); 

    // Get the dataset/dataspace dimensions
    hsize_t num_timesteps;
    H5Sget_simple_extent_dims(time_dataspace, &num_timesteps, NULL);

    // Data buffer to return 
    std::cout << "Number of timesteps we are reserving memory for = " << num_timesteps << "\n"; 
    std::vector<double> ret(num_timesteps);         

    PrintMemoryUsage("made memory space"); 

    // Read data from hyperslab in the file into the hyperslab in memory 
    H5Dread(mUnlimitedDatasetId, 
            H5T_NATIVE_DOUBLE, 
            H5S_ALL, 
            H5S_ALL, 
            H5P_DEFAULT, 
            &ret[0]); 

    PrintMemoryUsage("read into memory space"); 

    H5Sclose(time_dataspace); 

    return ret; 
}

而输出是

Number of timesteps we are reserving memory for = 400001
made memory space: memory use = 43.5898 MB.
read into memory space: memory use = 2182.4 MB.

(使用此代码诊断分配给程序的内存量 – 这看起来是否合理？：

#include <unistd.h>
#include <sys/resource.h>

void PrintMemoryUsage(const std::string& rPrefix)
{
    struct rusage rusage;
    getrusage( RUSAGE_SELF, &rusage );

    double max_res = (double)(rusage.ru_maxrss)/(1024);// Convert KB to MB
    std::cout << rPrefix << ": memory use = " << max_res <<  " MB.\n";
}

)

最佳答案 Yossarian的评论包含了线索 – 当我们检查主数据集的分块时,内存爆炸实际上来自一个仅包含时间值的相关1D数据集.后一个数据集的块大小为1.增加块大小解决了问题.

点赞