Searched refs:set_enable_all_reduce_fusion (Results 1 – 6 of 6) sorted by relevance
26 auto_parallel_context().set_enable_all_reduce_fusion(enable_all_reduce_fusion=True)
23 auto_parallel_context().set_enable_all_reduce_fusion(enable_all_reduce_fusion=True)
110 auto_parallel_context().set_enable_all_reduce_fusion(enable_all_reduce_fusion=True)
111 void set_enable_all_reduce_fusion(bool enable_all_reduce_fusion) { in set_enable_all_reduce_fusion() function
444 def set_enable_all_reduce_fusion(self, enable_all_reduce_fusion): member in _AutoParallelContext454 self._context_handle.set_enable_all_reduce_fusion(enable_all_reduce_fusion)
163 .def("set_enable_all_reduce_fusion", &ParallelContext::set_enable_all_reduce_fusion, in PYBIND11_MODULE()