SWDEV-299893 - Set preferred node affinity
Set affinity to the node nearest to default GPU at init. Afterthat
set it to NUMA node thats nearest to whatever GPU is set with
hipSetDevice
Change-Id: I85749258ea7c25385096ffe4089a70c948f332c7
Change-Id: I99a92c922655e22955bee512073b6ac8e6ced3a2
[ROCm/clr commit: 0d5bc833c3]
This commit is contained in:
@@ -84,6 +84,8 @@ Device* getCurrentDevice() {
|
||||
void setCurrentDevice(unsigned int index) {
|
||||
assert(index<g_devices.size());
|
||||
g_device = g_devices[index];
|
||||
uint32_t preferredNumaNode = g_device->devices()[0]->getPreferredNumaNode();
|
||||
amd::Os::setPreferredNumaNode(preferredNumaNode);
|
||||
}
|
||||
|
||||
amd::HostQueue* getQueue(hipStream_t stream) {
|
||||
|
||||
Reference in New Issue
Block a user