GPU allocation for CPX Unit Tests using PCI bus id (#1403)

* mapping devices wrt pci

* Gpu allocation by using pci mapping

* Passing gpuPriorityOrder in as an argument rather than making the functions non-static.

* Removing redundant testBed instance calling
Ta zmiana jest zawarta w:
saurabhAMD authored and GitHub committed 2024-11-04 10:51:00 -06:00
1 rodzic 1c45962273
złożenie 69b2b712ab
9 plików zmienionych
+119 -19

Żadne pliki nie pasują do Twojego wyszukiwania

+2 -1
Wyświetl plik
@@ -34,7 +34,8 @@ namespace RcclUnitTesting
{
int const numProcesses = isMultiProcess ? totalRanks : 1;
// Initialize communicators in non-blocking mode
testBed.InitComms(TestBed::GetDeviceIdsList(numProcesses, totalRanks), 1, 1, 1, useBlocking);
const std::vector<int>& gpuPriorityOrder = testBed.ev.GetGpuPriorityOrder();
testBed.InitComms(TestBed::GetDeviceIdsList(numProcesses, totalRanks, gpuPriorityOrder), 1, 1, 1, useBlocking);
// Loop over various collective functions
for (auto funcType : funcTypes)