AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-369792 - Fix to use the roclr context already created when creating hip::Device rather than creating a new context in hip

Browse Source 
Change-Id: I6dbccb1a17dee0e9a72ec52cb1a5bb096315fa0f
This commit is contained in:
Sourabh Betigeri
2023-10-03 14:52:12 -07:00
committed by Sourabh Betigeri
parent fe7b36f3cb
commit 01aa3557f4
1 changed files with 6 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_context.cpp
+6 -17
View File
@@ -51,26 +51,15 @@ void init(bool* status) {
const std::vector<amd::Device*>& devices = amd::Device::getDevices(CL_DEVICE_TYPE_GPU, false);
for (unsigned int i=0; i<devices.size(); i++) {
const std::vector<amd::Device*> device(1, devices[i]);
amd::Context* context = new amd::Context(device, amd::Context::Info());
if (!context) {
// Enable active wait on the device by default
devices[i]->SetActiveWait(true);
// use the eternal contexts that already exist for new hip::Device's here
auto device = new Device(&devices[i]->context(), i);
if ((device == nullptr) || !device->Create()) {
*status = false;
return;
}
// Enable active wait on the device by default
devices[i]->SetActiveWait(true);
if (context && CL_SUCCESS != context->create(nullptr)) {
context->release();
} else {
auto device = new Device(context, i);
if ((device == nullptr) || !device->Create()) {
*status = false;
return;
}
g_devices.push_back(device);
}
g_devices.push_back(device);
}
amd::Context* hContext = new amd::Context(devices, amd::Context::Info());