Add new Rome model (#1304)

* Add another rome model and override

* Fix bug

* Fix typo

* Add ring

* Update ring

* Fix model matching

* Clean up

* Clean up

* Reverse rings for NCCL_RINGS input

* Only reverse NCCL_RINGS for ring graph

* Fix mapping issue when using  NCCL_RINGS

* Add NCCL_RINGS_REMAP to handle inconsistant net names

[ROCm/rccl commit: 532b70afb6]
This commit is contained in:
Wenkai Du
2024-08-23 08:45:43 +08:00
committed by GitHub
parent 886b576722
commit 157cc5f6ba
7 changed files with 46 additions and 10 deletions
+1 -1
View File
@@ -21,7 +21,7 @@
DIR="$(cd -P "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
for i in {0..82}
for i in {0..87}
do
if [[ $i -eq 50 ]] || [[ $i -eq 51 ]]
then