Home
last modified time | relevance | path

Searched defs:nvlMesh (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/distributed/c10d/
Dintra_node_comm.cpp52 static std::ostream& operator<<(std::ostream& os, const NvlMesh& nvlMesh) { in operator <<()
82 NvlMesh nvlMesh = {}; in getNvlMesh() local
159 static bool isHybridCubeMesh(const NvlMesh nvlMesh) { in isHybridCubeMesh()
180 static Topology detectTopology(const NvlMesh nvlMesh, size_t worldSize) { in detectTopology()
329 auto nvlMesh = getNvlMesh(rankToBusId); in rendezvous() local
Dintra_node_comm.cu304 std::optional<HybridCubeMesh> getHybridCubeMesh(NvlMesh nvlMesh) { in getHybridCubeMesh()
501 void* initTopoInfo(Topology topology, NvlMesh nvlMesh, size_t rank) { in initTopoInfo()