From 2c859b045960477f63ab4f27e5105a1a6945b86d Mon Sep 17 00:00:00 2001 From: mengpenghui Date: Mon, 28 Nov 2022 15:29:25 +0800 Subject: [PATCH] fix error of torch_mlu --- docs/en/understand_mmcv/ops.md | 2 +- docs/zh_cn/understand_mmcv/ops.md | 2 +- tests/test_ops/test_modulated_deform_conv.py | 2 -- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/docs/en/understand_mmcv/ops.md b/docs/en/understand_mmcv/ops.md index 8a4bfe88eb..822ee15589 100644 --- a/docs/en/understand_mmcv/ops.md +++ b/docs/en/understand_mmcv/ops.md @@ -32,7 +32,7 @@ We implement common ops used in detection, segmentation, etc. | MaskedConv | | √ | √ | | √ | | MergeCells | | √ | | | | | MinAreaPolygon | | √ | | | | -| ModulatedDeformConv2d | √ | √ | | | √ | +| ModulatedDeformConv2d | √ | √ | √ | | √ | | MultiScaleDeformableAttn | | √ | √ | | | | NMS | √ | √ | √ | | √ | | NMSRotated | √ | √ | | | | diff --git a/docs/zh_cn/understand_mmcv/ops.md b/docs/zh_cn/understand_mmcv/ops.md index 7cf062ef79..23d9b6e5fd 100644 --- a/docs/zh_cn/understand_mmcv/ops.md +++ b/docs/zh_cn/understand_mmcv/ops.md @@ -32,7 +32,7 @@ MMCV 提供了检测、分割等任务中常用的算子 | MaskedConv | | √ | √ | | √ | | MergeCells | | √ | | | | | MinAreaPolygon | | √ | | | | -| ModulatedDeformConv2d | √ | √ | | | √ | +| ModulatedDeformConv2d | √ | √ | √ | | √ | | MultiScaleDeformableAttn | | √ | √ | | | | NMS | √ | √ | √ | | √ | | NMSRotated | √ | √ | | | | diff --git a/tests/test_ops/test_modulated_deform_conv.py b/tests/test_ops/test_modulated_deform_conv.py index 035d3b4e54..6bf7cb8b04 100644 --- a/tests/test_ops/test_modulated_deform_conv.py +++ b/tests/test_ops/test_modulated_deform_conv.py @@ -44,8 +44,6 @@ def _test_mdconv(self, device='mlu' if IS_MLU_AVAILABLE else 'cuda'): if not torch.cuda.is_available() and device == 'cuda': pytest.skip('test requires GPU') - if not torch.mlu.is_available() and device == 'mlu': - pytest.skip('test requires MLU') if device == 'mlu': from mmcv.ops import \ ModulatedDeformConv2dPack_MLU as ModulatedDeformConv2dPack