From 258c868a4951ccaaba6bb4c67e0a3de412f69dc8 Mon Sep 17 00:00:00 2001 From: Evgeny Date: Thu, 5 Sep 2019 20:20:38 -0500 Subject: [PATCH] setting hip backend library by env HIP_BACKEND_LIB for VDI support [ROCm/roctracer commit: 3aad90a32073f0b456c089d9d51a644817310f56] --- projects/roctracer/src/core/loader.h | 1 + projects/roctracer/src/core/roctracer.cpp | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/projects/roctracer/src/core/loader.h b/projects/roctracer/src/core/loader.h index bf3530c2ab..dfddb11a20 100644 --- a/projects/roctracer/src/core/loader.h +++ b/projects/roctracer/src/core/loader.h @@ -42,6 +42,7 @@ class BaseLoader : public T { } static loader_t* GetRef() { return instance_; } + static void SetLibName(const char *name) { lib_name_ = name; } private: BaseLoader(bool preload) { diff --git a/projects/roctracer/src/core/roctracer.cpp b/projects/roctracer/src/core/roctracer.cpp index 7c38dcc3b3..8fd429448b 100644 --- a/projects/roctracer/src/core/roctracer.cpp +++ b/projects/roctracer/src/core/roctracer.cpp @@ -1048,8 +1048,11 @@ PUBLIC_API roctracer_status_t roctracer_set_properties( break; } case ACTIVITY_DOMAIN_HCC_OPS: - case ACTIVITY_DOMAIN_HIP_API: + case ACTIVITY_DOMAIN_HIP_API: { + const char* hip_backend_lib_name = getenv("HIP_BACKEND_LIB"); + if (hip_backend_lib_name != NULL) roctracer::HccLoader::Instance().SetLibName(hip_backend_lib_name); mark_api_callback_ptr = reinterpret_cast(properties); + } default: EXC_RAISING(ROCTRACER_STATUS_BAD_DOMAIN, "invalid domain ID(" << domain << ")"); }