fix a bug when initializing a vector of hipFunction_t (#1949)

[ROCm/clr commit: a2281268ba]
This commit is contained in:
zhaozhangjian
2020-03-17 16:35:07 +08:00
committed by GitHub
parent 4d9df4192e
commit 02ab8feabd
+2 -2
View File
@@ -50,7 +50,7 @@ __hipRegisterFatBinary(const void* data)
return nullptr; return nullptr;
} }
auto modules = new std::vector<hipModule_t>{g_deviceCnt}; auto modules = new std::vector<hipModule_t>(g_deviceCnt);
if (!modules) { if (!modules) {
return nullptr; return nullptr;
} }
@@ -133,7 +133,7 @@ extern "C" void __hipRegisterFunction(
int* wSize) int* wSize)
{ {
HIP_INIT_API(NONE, modules, hostFunction, deviceFunction, deviceName); HIP_INIT_API(NONE, modules, hostFunction, deviceFunction, deviceName);
std::vector<hipFunction_t> functions{g_deviceCnt}; std::vector<hipFunction_t> functions(g_deviceCnt);
assert(modules && modules->size() >= g_deviceCnt); assert(modules && modules->size() >= g_deviceCnt);
for (int deviceId = 0; deviceId < g_deviceCnt; ++deviceId) { for (int deviceId = 0; deviceId < g_deviceCnt; ++deviceId) {