adding hipKernelNameRefByPtr function

Change-Id: Iefc18967b10394b85a207ffdb5bbfe5e3601474d
Tá an tiomantas seo le fáil i:
Evgeny
2020-05-28 06:59:56 -05:00
tiomanta ag Evgeny Shcherbakov
tuismitheoir eb31e45123
tiomantas d863edb8ba
D'athraigh 3 comhad le 16 breiseanna agus 0 scriosta
+1
Féach ar an gComhad
@@ -3656,6 +3656,7 @@ hipError_t hipRegisterActivityCallback(uint32_t id, void* fun, void* arg);
hipError_t hipRemoveActivityCallback(uint32_t id);
const char* hipApiName(uint32_t id);
const char* hipKernelNameRef(const hipFunction_t f);
const char* hipKernelNameRefByPtr(const void* hostFunction, hipStream_t stream);
#ifdef __cplusplus
} /* extern "C" */
#endif
+1
Féach ar an gComhad
@@ -181,6 +181,7 @@ global:
hipRemoveActivityCallback;
hipApiName;
hipKernelNameRef;
hipKernelNameRefByPtr;
hipProfilerStart;
hipProfilerStop;
hiprtcCompileProgram;
+14
Féach ar an gComhad
@@ -19,6 +19,7 @@
THE SOFTWARE. */
#include "hip/hip_runtime.h"
#include "hip_internal.hpp"
#include "hip_prof_api.h"
// HIP API callback/activity
@@ -27,6 +28,19 @@ api_callbacks_table_t callbacks_table;
extern const std::string& FunctionName(const hipFunction_t f);
const char* hipKernelNameRef(const hipFunction_t f) { return FunctionName(f).c_str(); }
const char* hipKernelNameRefByPtr(const void *hostFunction, hipStream_t stream) {
hip::Stream* s = reinterpret_cast<hip::Stream*>(stream);
int deviceId = (s != nullptr)? s->DeviceId() : ihipGetDevice();
if (deviceId == -1) {
DevLogPrintfError("Wrong Device Id: %d \n", deviceId);
return NULL;
}
hipFunction_t func = PlatformState::instance().getFunc(hostFunction, deviceId);
if (func == nullptr) {
return NULL;
}
return hipKernelNameRef(func);
}
hipError_t hipRegisterApiCallback(uint32_t id, void* fun, void* arg) {
return callbacks_table.set_callback(id, reinterpret_cast<api_callbacks_table_t::fun_t>(fun), arg) ?