Adding rocprofilerv2
Change-Id: Ic0cc280ba207d2b8f6ccae1cd4ac3184152fc1ad
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
#include "../common/common.h"
|
||||
#include <chrono>
|
||||
#include <thread>
|
||||
#include <iostream>
|
||||
|
||||
#include "rocprofiler.h"
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
int poll_duration = 5;
|
||||
if (argc > 1) poll_duration = atoi(argv[1]);
|
||||
|
||||
CHECK_ROCPROFILER(rocprofiler_initialize());
|
||||
printf("initialize\n");
|
||||
|
||||
rocprofiler_session_id_t dp_session_id;
|
||||
std::vector<const char*> counters;
|
||||
counters.emplace_back("FETCH_SIZE");
|
||||
|
||||
printf("session create\n");
|
||||
|
||||
int gpu_agent = 0;
|
||||
int cpu_agent = 0;
|
||||
CHECK_ROCPROFILER(rocprofiler_device_profiling_session_create(&counters[0], counters.size(),
|
||||
&dp_session_id, gpu_agent, cpu_agent));
|
||||
|
||||
printf("session start \n");
|
||||
// start GPU device profiling
|
||||
CHECK_ROCPROFILER(rocprofiler_device_profiling_session_start(dp_session_id));
|
||||
|
||||
using std::chrono::duration_cast;
|
||||
using std::chrono::high_resolution_clock;
|
||||
using std::chrono::milliseconds;
|
||||
|
||||
auto t1 = high_resolution_clock::now();
|
||||
|
||||
do {
|
||||
printf("polling\n");
|
||||
std::vector<rocprofiler_device_profile_metric_t> data(counters.size());
|
||||
// Poll metrics
|
||||
CHECK_ROCPROFILER(rocprofiler_device_profiling_session_poll(dp_session_id, &data[0]));
|
||||
|
||||
for (size_t i = 0; i < data.size(); i++)
|
||||
std::cout << data[i].metric_name << ": " << data[i].value.value << std::endl;
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(1));
|
||||
// break;
|
||||
} while (--poll_duration > 0);
|
||||
|
||||
auto t2 = high_resolution_clock::now();
|
||||
/* Getting number of milliseconds as an integer. */
|
||||
auto ms_int = duration_cast<milliseconds>(t2 - t1);
|
||||
|
||||
std::cout << ms_int.count() << "ms\n";
|
||||
|
||||
// Stop session
|
||||
CHECK_ROCPROFILER(rocprofiler_device_profiling_session_stop(dp_session_id));
|
||||
|
||||
// Destroy session
|
||||
CHECK_ROCPROFILER(rocprofiler_device_profiling_session_destroy(dp_session_id));
|
||||
|
||||
return 0;
|
||||
}
|
||||
Viittaa uudesa ongelmassa
Block a user