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

SWDEV-488290 - Remove Stream to Engine logic and rely on engine query status HSA API.

Browse Source 
Change-Id: I469ab6679360c8ee8d4ee515678a8aa8d4578ebf


[ROCm/clr commit: cc62a82347]
This commit is contained in:
kjayapra-amd
2025-01-31 07:00:02 -05:00
committed by Karthik Jayaprakash
parent cb281e23cd
commit 892d7bb064
3 changed files with 21 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/rocclr/device/rocm/rocdevice.cpp
-39
View File
@@ -1240,11 +1240,6 @@ bool Device::populateOCLDeviceConstants() {
return false;
}
for (uint32_t i = 0; i < info_.numSDMAengines_; i++) {
engineAssignMap_[1 << i] = 0;
}
checkAtomicSupport();
assert(system_segment_.handle != 0);
@@ -3515,46 +3510,12 @@ void Device::HiddenHeapInit(const VirtualGPU& gpu) {
std::call_once(heap_initialized_, HeapZeroOut);
}
// ================================================================================================
uint32_t Device::fetchSDMAMask(const device::BlitManager* handle, bool readEngine) const {
uint32_t engine = 0;
{
amd::ScopedLock lock(vgpusAccess());
for (auto it = engineAssignMap_.rbegin(); it != engineAssignMap_.rend(); ++it) {
// If blitManager handle is in the map return the engine ID else
// add to the map
if (it->second == handle) {
engine = it->first;
break;
} else if (it->second == 0) {
it->second = handle;
engine = it->first;
break;
}
}
}
return (readEngine ? maxSdmaReadMask_ : maxSdmaWriteMask_) & engine;
}
// ================================================================================================
void Device::getSdmaRWMasks(uint32_t* readMask, uint32_t* writeMask) const {
*readMask = maxSdmaReadMask_;
*writeMask = maxSdmaWriteMask_;
}
// ================================================================================================
void Device::resetSDMAMask(const device::BlitManager* handle) const {
amd::ScopedLock lock(vgpusAccess());
for (auto& it : engineAssignMap_) {
if (it.second == handle) {
it.second = 0;
break;
}
}
}
// ================================================================================================
void Device::AddKernel(Kernel& gpuKernel) const {
amd::ScopedLock lock(vgpusAccess());