Skipping AllReduce test on more than 8 ranks for FP8 type on Hyabusa (#1598)
* Skipping AllReduce FP8 test on 9 to 16 ranks (gfx90a) as it's using Tree algorithm not RING
---------
Co-authored-by: Marzieh Berenjkoub <mberenjk@amd.com>
[ROCm/rccl commit: 5f691aaf65]
Tá an tiomantas seo le fáil i:
1 tuismitheoir
74d6537141
tiomantas
a3a598efb3
3 comhad athraithe
+12
Níor tháinig aon chomhaid le do chuardach
@@ -681,6 +681,14 @@ namespace RcclUnitTesting
|
||||
|
||||
for (int ftIdx = 0; ftIdx < funcTypes.size() && isCorrect; ++ftIdx)
|
||||
for (int dtIdx = 0; dtIdx < dataTypes.size() && isCorrect; ++dtIdx)
|
||||
{
|
||||
//Skipping AllReduce FP8 test on 9 to 16 ranks (gfx90a).
|
||||
if(ev.isGfx90 && numRanks > 8 && funcTypes[ftIdx] == ncclCollAllReduce
|
||||
&& (dataTypes[dtIdx] == ncclFp8E4M3
|
||||
|| dataTypes[dtIdx] == ncclFp8E5M2))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
for (int rdIdx = 0; rdIdx < redOps.size() && isCorrect; ++rdIdx)
|
||||
for (int rtIdx = 0; rtIdx < roots.size() && isCorrect; ++rtIdx)
|
||||
for (int ipIdx = 0; ipIdx < inPlaceList.size() && isCorrect; ++ipIdx)
|
||||
@@ -764,6 +772,7 @@ namespace RcclUnitTesting
|
||||
}
|
||||
this->DeallocateMem();
|
||||
}
|
||||
}
|
||||
this->DestroyComms();
|
||||
}
|
||||
}
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir