Add code-object-v3 support for the HIP-Clang path.

[ROCm/clr commit: 74578160bd]
Этот коммит содержится в:
Laurent Morichetti
2019-02-05 14:43:26 -08:00
родитель e5d45862bb
Коммит 0ef706ff97
+6 -2
Просмотреть файл
@@ -132,8 +132,12 @@ extern "C" void __hipRegisterFunction(
assert(modules && modules->size() >= g_deviceCnt);
for (int deviceId = 0; deviceId < g_deviceCnt; ++deviceId) {
hipFunction_t function;
if (hipSuccess == hipModuleGetFunction(&function, modules->at(deviceId), deviceName) &&
function != nullptr) {
if ((hipSuccess == hipModuleGetFunction(&function, modules->at(deviceId), deviceName) ||
// With code-object-v3, we need to match the kernel descriptor symbol name
(hipSuccess == hipModuleGetFunction(
&function, modules->at(deviceId),
(std::string(deviceName) + std::string(".kd")).c_str()
))) && function != nullptr) {
functions[deviceId] = function;
}
else {