utils.h

Go to the documentation of this file.

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 * Distributed under BSD 3-Clause license.                                   *
 * Copyright by The HDF Group.                                               *
 * Copyright by the Illinois Institute of Technology.                        *
 * All rights reserved.                                                      *
 *                                                                           *
 * This file is part of Hermes. The full Hermes copyright notice, including  *
 * terms governing use, modification, and redistribution, is contained in    *
 * the COPYING file, which can be found at the top directory. If you do not  *
 * have access to the file, you may request a copy from help@hdfgroup.org.   *
 * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
#ifndef HERMES_UTILS_H_
#define HERMES_UTILS_H_
 
#include <assert.h>
 
#include <chrono>
#include <map>
 
#include "hermes_types.h"
 
#if HERMES_ENABLE_TIMING
 
#define HERMES_BEGIN_TIMED_BLOCK(func_name)                                   \
  auto hermes_timed_block_start_ = std::chrono::high_resolution_clock::now(); \
  const char *hermes_timed_block_func_name_ = (func_name);
 
// TODO(chogan): Probably want to prepend rank to output
#define HERMES_END_TIMED_BLOCK()                                            \
  auto hermes_timed_block_end_ = std::chrono::high_resolution_clock::now(); \
  auto hermes_timed_block_elapsed_ =                                        \
      hermes_timed_block_end_ - hermes_timed_block_start_;                  \
  double hermes_timed_block_seconds_ =                                      \
      std::chrono::duration<double>(hermes_timed_block_elapsed_).count();   \
  VLOG(1) << hermes_timed_block_func_name_ << " took "                      \
          << hermes_timed_block_seconds_ << " seconds\n";
 
#else
#define HERMES_BEGIN_TIMED_BLOCK(func_name) 
#define HERMES_END_TIMED_BLOCK()            
#endif                                      // HERMES_ENABLE_TIMING
 
namespace hermes {
 
size_t RoundUpToMultiple(size_t val, size_t multiple);
 
size_t RoundDownToMultiple(size_t val, size_t multiple);
 
void InitDefaultConfig(Config *config);
 
void FailedLibraryCall(std::string func);
 
namespace testing {
enum class BlobSizeRange {
  kSmall,
  kMedium,
  kLarge,
  kXLarge,
  kHuge,
};
 
struct TargetViewState {
  std::vector<hermes::u64> bytes_capacity;  
  std::vector<hermes::u64> bytes_available; 
  std::vector<hermes::f32> bandwidth;       
  int num_devices;                          
};
 
TargetViewState InitDeviceState(u64 total_target = 4, bool homo_dist = true);
 
u64 UpdateDeviceState(PlacementSchema &schema, TargetViewState &node_state);
 
void PrintNodeState(TargetViewState &node_state);
 
std::vector<TargetID> GetDefaultTargets(size_t n);
 
std::vector<size_t> GenFixedTotalBlobSize(size_t total_size,
                                          BlobSizeRange range);
 
}  // namespace testing
}  // namespace hermes
#endif  // HERMES_UTILS_H_