onert::exec::RawMinMaxDumper Class Reference

#include <MinMaxData.h>

Public Member Functions
	RawMinMaxDumper (const std::string &filename)
void	dump (const exec::IOMinMaxMap &in_minmax, const exec::OpMinMaxMap &op_minmax) const
	Dump input minmax map.

Detailed Description

Definition at line 55 of file MinMaxData.h.

Constructor & Destructor Documentation

RawMinMaxDumper()

onert::exec::RawMinMaxDumper::RawMinMaxDumper

(

const std::string &

filename

)

Definition at line 26 of file MinMaxData.cc.

: _filename(filename) {}

Member Function Documentation

dump()

void onert::exec::RawMinMaxDumper::dump	(	const exec::IOMinMaxMap &	in_minmax,
		const exec::OpMinMaxMap &	op_minmax
	)		const

Dump input minmax map.

Parameters

[in]	in_minmax	input minmax map
[in]	op_minmax	op minmax map

Definition at line 28 of file MinMaxData.cc.

{
  // Find file is already exist for modifying
  auto file = std::fopen(_filename.c_str(), "rb+");
  uint32_t runs = 1;
 
  // Magic code and version
  // Match with runtime/onert/odc/MinMaxReader.cc
  // TODO Use util to share code and version
  const uint32_t MAGIC_CODE = 0x4F4D4D44;
  const uint32_t VERSION = 1;
  if (!file)
  {
    // If file is not exist, create new file
    file = std::fopen(_filename.c_str(), "wb+");
    if (!file)
      throw std::runtime_error{"RawMinMaxDumper: Failed to open minmax file " + _filename};
 
    // Write magic code and version
    std::fwrite(&MAGIC_CODE, sizeof(uint32_t), 1, file);
    std::fwrite(&VERSION, sizeof(uint32_t), 1, file);
  }
  else
  {
    // Check magic code and version
    std::fseek(file, 0, SEEK_SET);
    uint32_t read_magic_code = 0;
    uint32_t read_version = 0;
    bool rewrite = true;
    if (std::fread(&read_magic_code, sizeof(uint32_t), 1, file) == 1 &&
        read_magic_code == MAGIC_CODE &&
        std::fread(&read_version, sizeof(uint32_t), 1, file) == 1 && read_version == VERSION)
      rewrite = false;
 
    // Destroy and create if file is not valid
    if (rewrite)
    {
      std::fclose(file);
      file = std::fopen(_filename.c_str(), "wb+");
      if (!file)
        throw std::runtime_error{"RawMinMaxDumper: Failed to rewrite minmax file " + _filename};
 
      // Write magic code and version
      std::fwrite(&MAGIC_CODE, sizeof(uint32_t), 1, file);
      std::fwrite(&VERSION, sizeof(uint32_t), 1, file);
    }
  }
 
  // Read run count
  if (std::fread(&runs, sizeof(uint32_t), 1, file) == 1)
    runs++;
  else
    runs = 1;
 
  // TODO Verify file size
 
  // Overwrite run count
  std::fseek(file, sizeof(MAGIC_CODE) + sizeof(VERSION), SEEK_SET);
  std::fwrite(&runs, sizeof(uint32_t), 1, file);
 
  // Go to end of file to append new data
  std::fseek(file, 0, SEEK_END);
 
  uint32_t input_count = input_minmax.size();
  uint32_t op_count = op_minmax.size();
 
  // Write op_count and input_count
  std::fwrite(&op_count, sizeof(uint32_t), 1, file);
  std::fwrite(&input_count, sizeof(uint32_t), 1, file);
 
  // For each op
  for (auto &&[index, minmax] : op_minmax)
  {
    const uint32_t model_idx = 0;
    const uint32_t subg_idx = index.first.value();
    const uint32_t op_idx = index.second.value();
 
    // Write model/subg/op index
    std::fwrite(&model_idx, sizeof(uint32_t), 1, file);
    std::fwrite(&subg_idx, sizeof(uint32_t), 1, file);
    std::fwrite(&op_idx, sizeof(uint32_t), 1, file);
 
    // Write min/max
    std::fwrite(minmax.data, sizeof(float), 2, file);
  }
 
  // For each input
  for (auto &&[index, minmax] : input_minmax)
  {
    const uint32_t model_idx = 0;
    const uint32_t subg_idx = index.first.value();
    const uint32_t input_idx = index.second.value();
 
    // Write model/subg/input index
    std::fwrite(&model_idx, sizeof(uint32_t), 1, file);
    std::fwrite(&subg_idx, sizeof(uint32_t), 1, file);
    std::fwrite(&input_idx, sizeof(uint32_t), 1, file);
 
    // Write min/max
    std::fwrite(minmax.data, sizeof(float), 2, file);
  }
 
  std::fclose(file);
}

References onert::util::MinMaxMap< N, Hash >::size().

---

The documentation for this class was generated from the following files:

• runtime/onert/core/src/exec/MinMaxData.h
• runtime/onert/core/src/exec/MinMaxData.cc