graph_utils.h

#ifndef __GRAPH_UTILS_H__
#define __GRAPH_UTILS_H__
 
#include <adf.h>
 
// for 16384B (note 32768B DDR per AIE)
#define TILE_BYTES        32768
#define MAX_PARAM_BYTES   16384
#define MAX_FLOAT_PARAMS  MAX_PARAM_BYTES / sizeof(float)
 
#ifdef EXTERNAL_IO
#define PLIO64_ARG(TXT_PATH) adf::plio_64_bits
#else
#define PLIO64_ARG(TXT_PATH) adf::plio_64_bits, TXT_PATH
#endif
 
#define adfCheck(stmt, log_name)          \
  do {                                    \
    adf::return_code ret = stmt;          \
    if (ret != adf::ok)                   \
      printf("ERROR [%s]: failed\n", log_name); \
  } while (0)
 
 
// uses 2/2 performance counters
void get_graph_throughput_by_port(
  adf::graph graph,
  const std::string& log_name,
  adf::IoAttr& plio_port, 
  const size_t window_len, 
  const size_t sample_bytesize,
  const size_t iter_cnt
) {
  adf::event::handle ehdl = adf::event::start_profiling(plio_port, adf::event::io_stream_start_to_bytes_transferred_cycles, window_len * iter_cnt * sample_bytesize);
 
  adfCheck(graph.run(iter_cnt), "run graph");
  adfCheck(graph.wait(), "wait graph");
 
  if (ehdl != adf::event::invalid_handle) {
    long long cycles = adf::event::read_profiling(ehdl);
    adf::event::stop_profiling(ehdl);
    printf("%s: Cycles %lld Throughput %f samples/s\n", log_name.c_str(), cycles, (double) window_len * iter_cnt / (cycles * 1e-9)); //samples/second
  } else {
    printf("%s: ERROR: Invalid handle. Only two performance counter in a AIE-PL interface tile. Event profile is not supported for x86sim.\n", log_name.c_str());
  }
}
 
// uses 2/2 performance counters, run first if multiple profiles are done sequentially as counters wait till net receives first data
void get_graph_latency(
  adf::graph graph,
  const std::string& log_name,
  adf::IoAttr& plio_input_port, 
  adf::IoAttr& plio_output_port,
  const size_t iter_cnt
) {
  adf::event::handle ehdl = adf::event::start_profiling(plio_input_port, plio_output_port, adf::event::io_stream_start_difference_cycles);
  
  adfCheck(graph.run(iter_cnt), "run graph");
  adfCheck(graph.wait(), "wait graph");
 
  if (ehdl != adf::event::invalid_handle) {
    long long cycles = adf::event::read_profiling(ehdl);
    adf::event::stop_profiling(ehdl); // performance counter is released and cleared 
    printf("[%s]: AIE Latency: %lld cycles, %fs (@1Ghz)\n", log_name.c_str(), cycles, (double) cycles * 1e-9);
  } else {
    printf("%s: ERROR: Invalid handle. Only two performance counter in a AIE-PL interface tile. Event profile is not supported for x86sim.\n", log_name.c_str());
  }
}
 
// Only useful for system test, uses 2/2 performance counters
void get_graph_bandwidth_by_port(
  adf::graph graph,
  const std::string& log_name,
  adf::IoAttr& plio_input_port, 
  adf::IoAttr& plio_output_port, 
  const size_t input_window_len,
  const size_t output_window_len,
  const size_t iter_cnt
) {
  adf::event::handle in_ehdl = adf::event::start_profiling(plio_input_port, adf::event::io_total_stream_running_to_idle_cycles);
  adf::event::handle out_ehdl = adf::event::start_profiling(plio_output_port, adf::event::io_total_stream_running_to_idle_cycles);
  
  adfCheck(graph.run(iter_cnt), "run graph");
  adfCheck(graph.wait(), "wait graph");
 
  long long cycles;
  if (in_ehdl != adf::event::invalid_handle) {
    cycles = adf::event::read_profiling(in_ehdl);
    adf::event::stop_profiling(in_ehdl);
    printf("%s: Cycles %lld Bandwidth %f, ==1: graph faster than PLIO, < 1: input port stalled\n", log_name.c_str(), cycles, (double) input_window_len * iter_cnt / cycles);
  } else {
    printf("%s: ERROR: Invalid handle. Only two performance counter in a AIE-PL interface tile. Event profile is not supported for x86sim.\n", log_name.c_str());
  }
 
  if (out_ehdl != adf::event::invalid_handle) {
    cycles = adf::event::read_profiling(out_ehdl);
    adf::event::stop_profiling(out_ehdl);
    printf("%s: Cycles %lld Bandwidth %f, ==1: graph faster than PLIO, < 1: output port stalled\n", log_name.c_str(), cycles, (double) output_window_len * iter_cnt / cycles);
  } else {
    printf("%s: ERROR: Invalid handle. Only two performance counter in a AIE-PL interface tile. Event profile is not supported for x86sim.\n", log_name.c_str());
  }
}
 
#endif // __GRAPH_UTILS_H__