memory.h

Go to the documentation of this file.

#ifndef MEMORY_H
#define MEMORY_H
 
#include <fstream>
#include <string>
 
namespace ModuleBase
{
 
class Memory
{
  public:
    Memory();
    ~Memory();
 
    static double record(const std::string &class_name,
                         const std::string &name,
                         const long &n,
                         const std::string &type,
                         const bool accumulate = false);
 
    static void record(
      const std::string &name_in,
      const size_t &n_in,
      const bool accumulate = false
    );
 
#if defined(__CUDA) || defined(__ROCM)
 
    static double record_gpu(const std::string &class_name,
                         const std::string &name,
                         const long &n,
                         const std::string &type,
                         const bool accumulate = false);
 
    static void record_gpu(
      const std::string &name_in,
      const size_t &n_in,
      const bool accumulate = false
    );
 
#endif
 
    static double &get_total(void)
    {
        return total;
    }
 
    static void finish(std::ofstream &ofs);
 
    static void print_all(std::ofstream &ofs);
 
    static void print(const int find_in);
 
    static double calculate_mem(const long &n, const std::string &type);
 
  private:
    static double total;
    static std::string *name;
    static std::string *class_name;
    static double *consume;
    static int n_memory;
    static int n_now;
    static bool init_flag;
 
#if defined(__CUDA) || defined(__ROCM)
    static double total_gpu;
    static std::string *name_gpu;
    static std::string *class_name_gpu;
    static double *consume_gpu;
    static int n_now_gpu;
    static bool init_flag_gpu;
#endif
 
    static int complex_matrix_memory; //(16 Byte)
    static int double_memory; //(8 Byte)
    static int int_memory; //(4 Byte)
    static int bool_memory;
    static int short_memory; //(2 Byte)
    static int float_memory; //(4 Byte)
};
 
} // namespace ModuleBase
 
#endif