Searched defs:fusion_threshold (Results 1 – 2 of 2) sorted by relevance
106 void ParallelContext::set_fusion_threshold_mb(int64_t fusion_threshold) { in set_fusion_threshold_mb()113 void ParallelContext::set_allgather_fusion_threshold_mb(int64_t fusion_threshold) { in set_allgather_fusion_threshold_mb()118 void ParallelContext::set_reducescatter_fusion_threshold_mb(int64_t fusion_threshold) { in set_reducescatter_fusion_threshold_mb()
171 def set_fusion_threshold_mb(self, fusion_threshold=64, comm_type="allreduce"): argument