utils.hpp

Go to the documentation of this file.

// Copyright Institute for Automotive Engineering (ika), RWTH Aachen University
// SPDX-License-Identifier: Apache-2.0
 
#pragma once
 
#include <algorithm>
#include <array>
#include <cstdint>
#include <cstring>
#include <functional>
#include <iostream>
#include <numeric>
#include <random>
#include <stdexcept>
#include <string>
#include <vector>
 
#include <common.h>
 
namespace triton_cpp {
 
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::vector<T>& data) {
  os << "[";
  if (data.size() > 0) {
    for (auto d = data.begin(); d != data.end() - 1; ++d) {
      os << *d << ", ";
    }
    os << data.back();
  }
  os << "]";
  return os;
};
 
template <typename It>
std::int64_t accumulate_shape(It begin, It end) {
  return std::accumulate(begin, end, 1l, [](std::int64_t a, std::int64_t b) { return a * std::max(b, 1l); });
}
 
inline void fail_on_error(const triton::client::Error& err, std::string message = "") {
  if (!err.IsOk()) {
    throw std::runtime_error(message + ": " + err.Message());
  }
}
 
inline std::string randstring(std::size_t len) {
  static constexpr auto chars =
      "0123456789"
      "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
      "abcdefghijklmnopqrstuvwxyz";
  thread_local static std::mt19937 rng{std::random_device{}()};
  thread_local static std::uniform_int_distribution<std::string::size_type> dist(0, std::strlen(chars) - 1);
 
  std::string result(len, '\0');
  std::generate_n(begin(result), len, [&]() { return chars[dist(rng)]; });
  return result;
}
 
}  // namespace triton_cpp