Searched refs:GetNcclCollectiveConfigForMlir (Results 1 – 4 of 4) sorted by relevance
78 NcclCollectiveConfig GetNcclCollectiveConfigForMlir(OpT op, in GetNcclCollectiveConfigForMlir() function
36 config.config = GetNcclCollectiveConfigForMlir(op, replica_count); in GetNcclAllGatherConfig()
38 config.config = GetNcclCollectiveConfigForMlir(op, replica_count); in GetNcclAllToAllConfig()
95 config.config = GetNcclCollectiveConfigForMlir(op, replica_count); in GetNcclAllReduceConfig()