enable tool reloading

This commit is contained in:
Evgeny
2020-03-14 00:13:22 -05:00
parent 3568ca4dbb
commit 1be273a1b4
5 changed files with 39 additions and 41 deletions
+8 -11
View File
@@ -585,19 +585,18 @@ void open_tracing_pool() {
void close_tracing_pool() {
if (roctracer_default_pool() != NULL) {
ROCTRACER_CALL(roctracer_flush_activity());
ROCTRACER_CALL(roctracer_close_pool());
}
}
// tool library is loaded
static bool is_loaded = false;
// tool unload method
void tool_unload() {
static bool is_unloaded = false;
ONLOAD_TRACE("begin, unloaded(" << is_unloaded << ")");
ONLOAD_TRACE("begin, loaded(" << is_loaded << ")");
if (is_unloaded == true) return;
is_unloaded = true;
roctracer_unload();
if (is_loaded == false) return;
is_loaded = false;
if (trace_roctx) {
ROCTRACER_CALL(roctracer_disable_domain_callback(ACTIVITY_DOMAIN_ROCTX));
@@ -627,12 +626,13 @@ void tool_unload() {
// tool load method
void tool_load() {
static bool is_loaded = false;
ONLOAD_TRACE("begin, loaded(" << is_loaded << ")");
if (is_loaded == true) return;
is_loaded = true;
roctracer::TraceBufferBase::StartWorkerThreadAll();
// Output file
const char* output_prefix = getenv("ROCP_OUTPUT_DIR");
if (output_prefix != NULL) {
@@ -817,9 +817,6 @@ void tool_load() {
printf(")\n");
}
roctracer::TraceBufferBase::StartWorkerThreadAll();
roctracer_load();
ONLOAD_TRACE_END();
}