AI/rocm-systems
2
0
Forkuj 0
Kod Zgłoszenia Oczekujące zmiany Actions Packages Projekty Wydania Wiki Aktywność

NCCL_TREES variable and rome model fixes (#856)

Przeglądaj źródła
This commit is contained in:
akolliasAMD
2023-08-21 10:35:37 -06:00
zatwierdzone przez GitHub
rodzic 148e3430f4
commit d33cd5a233
8 zmienionych plików z 58 dodań i 60 usunięć
Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files
src/graph/rome_models.h
+1
Wyświetl plik
@@ -23,6 +23,7 @@ THE SOFTWARE.
#define RCCL_ROME_MODELS_H_
ncclResult_t parseGraph(const char* str, struct ncclTopoSystem* system, struct ncclTopoGraph* graph, int* gpu_map, int* net_map);
ncclResult_t parseGraphLight(const char* str, struct ncclTopoSystem* system, struct ncclTopoGraph* graph, int* gpu_map);
ncclResult_t parseRome4P2H(struct ncclTopoSystem* system, struct ncclTopoGraph* graph);
ncclResult_t parseChordalRing(struct ncclTopoSystem* system, struct ncclTopoGraph* graph);
ncclResult_t parse1H16P(struct ncclTopoSystem* system, struct ncclTopoGraph* graph);