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
This commit is contained in:
Wenkai Du
2024-08-23 08:45:43 +08:00
committed by GitHub
parent db840f024e
commit 532b70afb6
7 changed files with 46 additions and 10 deletions
+1
View File
@@ -140,6 +140,7 @@ struct ncclTopoNode {
int collSupport;
int maxChannels;
int64_t busId;
int dev;
}net;
struct {
int arch;