2 Commits (dd5b415d2acfdc2ebd346f23c849556bfd2d0199)

Author	SHA1	Message	Date
wangjiangben-hw	a5db5f6682	[Feature] Support training on NPU device (#2262) ... * init npu * add npu extension and focal loss adapter * clean code * clean code * clean code * clean code * fix autocast bugs on npu (#2273) fix autocast bugs on npu (#2273) * code format * code format * code format * bug fix * pytorch_npu_helper.hpp clean code * Npu dev (#2306) * fix autocast bugs on npu * using scatter_kwargs in mmcv.device.scatter_gather * raise ImportError when compile with npu * add npu test case (#2307) * add npu test case * Update focal_loss.py * add comment * clean lint * update dtype assert * update DDP forward and comment * fix bug Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by: ckirchhoff <515629648@qq.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>	2022-09-30 21:05:37 +08:00
Zaida Zhou	6a03918f55	[Feature] Add support for mps (#2092) ... * [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	2022-07-07 16:05:49 +08:00