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

SWDEV-24806 - Added support for event wait and event record graph APIs

Browse Source 
hip graph implementation for APIs AddEventRecordNode, AddEventWaitNode, EventRecordNodeGetEvent, EventWaitNodeGetEvent, EventRecordNodeSetEvent,EventWaitNodeSetEvent, ExecEventRecordNodeSetEvent, ExecEventWaitNodeSetEvent

Change-Id: Ifde51f4591067795d6cbf5595372cbdbd8a7bc26
This commit is contained in:
anusha GodavarthySurya
2021-10-05 10:06:40 -07:00
parent c4eda85835
commit f6456c5e27
11 changed files with 353 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_module.cpp
+8 -3
View File
@@ -391,15 +391,20 @@ hipError_t ihipModuleLaunchKernel(hipFunction_t f, uint32_t globalWorkSizeX,
hip::Event* eStart = reinterpret_cast<hip::Event*>(startEvent);
hip::Event* eStop = reinterpret_cast<hip::Event*>(stopEvent);
command->enqueue();
if (startEvent != nullptr) {
eStart->addMarker(queue, command, false);
status = eStart->addMarker(hStream, command, false);
command->retain();
}
if (status != hipSuccess) {
return status;
}
if (stopEvent != nullptr) {
eStop->addMarker(queue, command, true);
status = eStop->addMarker(hStream, command, true);
command->retain();
}
if (status != hipSuccess) {
return status;
}
command->release();
return hipSuccess;