- 07 Jul, 2022 1 commit
-
-
Zaida Zhou authored
* [Feature] Add support for MPS * fix import error * update ut * fix error * trigger CI * use a unique basename for test file modules * avoid bc-breaking
-
- 15 Apr, 2022 1 commit
-
-
Qiza-lyhm authored
To have the same shape of MLU copy and Scatter.forward, a 1-dim list has to be added. Origin MMDP / MMDDP with this scatter_gather returns a list of tensor for each targets, and here MLU copy returns the same shape Tensor in a list of just one default target.
-
- 14 Apr, 2022 1 commit
-
-
Qiza-lyhm authored
* feat(MLU): Support PyTorch backend on MLU * MMCV support PyTorch backend on MLU * Add MLUDataParallel and MLUDistributedDataParallel * Add MLU operator support * [Fix]: Fix PR comments and add IS_MLU to get device available check * [Fix]: fix PR comments of dist_utils.py * [Doc] Rewrite annotations of functions. * [Docs] Rewrite annotation in distributed.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * [Docs] Fix lint Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
-
- 08 Aug, 2021 1 commit
-
-
Jerry Jiarui XU authored
* [Docs] Add header for files * change to OpenMMLab * add headers for .cpp, .cu, .h, .cuh * replace Open-MMLab with OpenMMLab
-
- 19 Aug, 2020 1 commit
-
-
Wenwei Zhang authored
* Fix mmcv-dataparallel * Fix (parallel): fix CPU inference with MMDataParallel * Update docstrings * Doc (parallel): refine docstrings * Fix (parallel): fix missing changes of train/val step function * resolve comments * Fix (data_parallel): fix bug when single gpu test return None
-
- 03 Jul, 2020 1 commit
-
-
Kai Chen authored
* update doc formats * update docstring
-
- 10 Jan, 2020 1 commit
-
-
Kai Chen authored
* add copyright header * change # to // for cpp files
-
- 08 Oct, 2018 1 commit
-
-
Kai Chen authored
-