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

[ROCm/hip commit: 6ab8711f02]
Cette révision appartient à :
Laurent Morichetti
2019-02-05 14:43:26 -08:00
Parent 08a20ec9fd
révision 1bbb9d455b
+6 -2
Voir le fichier
@@ -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 {