src/smpi/include/smpi_utils.hpp

   1 /* Copyright (c) 2016-2022. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #ifndef SMPI_UTILS_HPP
   7 #define SMPI_UTILS_HPP
   8 #include <xbt/base.h>
   9
  10 #include "smpi_f2c.hpp"
  11 #include "smpi_comm.hpp"
  12
  13 #include <cstddef>
  14 #include <string>
  15 #include <string_view>
  16 #include <vector>
  17
  18 // Methods used to parse and store the values for timing injections in smpi
  19 struct s_smpi_factor_t {
  20   size_t factor = 0;
  21   std::vector<double> values;
  22 };
  23
  24 namespace simgrid::smpi::utils {
  25
  26 class FactorSet {
  27   const std::string& name_;
  28   std::vector<s_smpi_factor_t> factors_;
  29   double default_value_;
  30   const std::function<double(std::vector<double> const&, double)> lambda_;
  31   bool initialized_ = false;
  32
  33 public:
  34   // Parse the factor from a string
  35   FactorSet(
  36       const std::string& name, double default_value = 1,
  37       std::function<double(std::vector<double> const&, double)> const& lambda = [](std::vector<double> const& values,
  38                                                                                    double) { return values.front(); });
  39   void parse(const std::string& string_values);
  40   bool is_initialized() const { return initialized_; }
  41   // Get the default value
  42   double operator()();
  43   // Get the factor to use for the provided size
  44   double operator()(double size);
  45 };
  46 XBT_PUBLIC void add_benched_time(double time);
  47 XBT_PUBLIC void account_malloc_size(size_t size, std::string_view file, int line, const void* ptr);
  48 XBT_PUBLIC void account_shared_size(size_t size);
  49 XBT_PUBLIC void print_time_analysis(double time);
  50 XBT_PUBLIC void print_buffer_info();
  51 XBT_PUBLIC void print_memory_analysis();
  52 XBT_PUBLIC void print_current_handle();
  53 XBT_PUBLIC void set_current_handle(F2C* handle);
  54 XBT_PUBLIC void set_current_buffer(int i, const char* name, const void* handle);
  55 XBT_PUBLIC size_t get_buffer_size(const void* ptr);
  56 XBT_PUBLIC void account_free(const void* ptr);
  57 XBT_PUBLIC int check_collectives_ordering(MPI_Comm comm, const std::string& call);
  58
  59 } // namespace simgrid::smpi::utils
  60 #endif