- Oct 24, 2022
-
-
wangjiangben-hw authored
* init npu * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/dist/dist.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * change to is_hccl_backend * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add comment with AmpOptimWrapper * Update mmengine/runner/amp.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/runner/amp.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add npu fn in base_model * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * clean lint * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/model/base_model/base_model.py Co-authored-by:
Mashiro <57566630+HAOCHENYE@users.noreply.github.com> * add is_npu_available * try to fix * Add comments * Refine grammar Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by:
Mashiro <57566630+HAOCHENYE@users.noreply.github.com> Co-authored-by:
HAOCHENYE <21724054@zju.edu.cn>
-
- Oct 08, 2022
-
-
Austin Welch authored
* Add smddp dist backend option * [Dev]: Upgrade pre commit hooks (#576) * Upgrade the versions of pre-commit-hooks * update zh-cn.yaml * [Docs] Fix the docstring of model sub-package (#573) * [Doc]: Update config.md (#562) * Update config.md * Update config.md * [Doc] delete the error comment in docs (#514) Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by:
Zhengfei-0311 <78833899+Zhengfei-0311@users.noreply.github.com> Co-authored-by:
vansin <msnode@163.com>
-
- Aug 15, 2022
-
-
Kai Hu authored
-
Zaida Zhou authored
-
- Jun 16, 2022
-
-
Jiazhen Wang authored
* support mlu * add ut and refine docstring
-
- May 10, 2022
-
-
Yining Li authored
-
- May 05, 2022
-
-
Zaida Zhou authored
-
- Apr 27, 2022
-
-
Zaida Zhou authored
* [Enhancement] Handle the device type of inputs in functions * rename and move three fucntions to dist/utils.py * minor refinement * rename dist to torch_dist in utils.py * update unit tests * refine unit tests * add unit tests * fix unit tests * replace Sequence with list and tuple * rename get_backend_device to get_comm_device * fix unit tests * fix unit tests * refactor and add more unit tests * cast_data_device does not support set type
-
- Mar 05, 2022
-
-
Zaida Zhou authored
* [Feature] Add distributed module * fix IS_DIST error * all_reduce_dict does operations in-place * support 'mean' operation * provide local group process * add tmpdir argument for collect_results * add unit tests * refactor unit tests * simplify steps to create multiple processes * minor fix * describe the different of *gather* in mmengine and pytorch * minor fix * add unit tests for nccl * test nccl backend in multiple gpu * add get_default_group function to handle different torch versions * minor fix * [Feature] Add distributed module * fix IS_DIST error * all_reduce_dict does operations in-place * support 'mean' operation * provide local group process * add tmpdir argument for collect_results * add unit tests * refactor unit tests * simplify steps to create multiple processes * minor fix * describe the different of *gather* in mmengine and pytorch * minor fix * add unit tests for nccl * test nccl backend in multiple gpu * add get_default_group function to handle different torch versions * minor fix * minor fix * handle torch1.5 * handle torch1.5 * minor fix * fix typo * refactor unit tests * nccl does not support gather and gather_object * fix gather * fix collect_results_cpu * fix collect_results and refactor unit tests * fix collect_results unit tests * handle torch.cat in torch1.5 * refine docstring * refine docstring * fix comments * fix comments
-