utils.h

#include <chrono>
#include <cstdlib>
#include <cuda.h>
#include <iostream>
#include <nlohmann/json.hpp>
#include <tuple>

using namespace std;
using json = nlohmann::json;

tuple<CUcontext, CUstream_st *> setup_cuda();
CUdeviceptr make_data(size_t size);
void deepcopyCUdeviceptr(CUdeviceptr src, CUdeviceptr *dst, size_t size);
void cleanup(CUcontext context, CUstream_st *stream, CUdeviceptr d_X);

struct test_result {
  std::chrono::duration<double> duration;
  int nreps;
  int grid_size;
  int block_size;

  json to_json() const {
    return json{{"block_size", block_size},
                {"grid_size", grid_size},
                {"nreps", nreps},
                {"duration",
                 std::chrono::duration_cast<std::chrono::nanoseconds>(duration)
                     .count()}};
  }
};