63 lines
3.5 KiB
Markdown
Raw Normal View History

## ops
We implement common ops used in detection, segmentation, etc.
| Device | CPU | CUDA | MLU | MPS |
| ---------------------------- | --- | ---- | --- | --- |
| ActiveRotatedFilter | √ | √ | | |
| AssignScoreWithK | | √ | | |
| BallQuery | | √ | | |
| BBoxOverlaps | | √ | √ | √ |
| BorderAlign | | √ | | |
| BoxIouRotated | √ | √ | | |
| BoxIouQuadri | √ | √ | | |
| CARAFE | | √ | √ | |
| ChamferDistance | | √ | | |
| CrissCrossAttention | | √ | | |
| ContourExpand | √ | | | |
| ConvexIoU | | √ | | |
| CornerPool | | √ | | |
| Correlation | | √ | | |
| Deformable Convolution v1/v2 | √ | √ | | |
| Deformable RoIPool | | √ | √ | |
| DiffIoURotated | | √ | | |
| DynamicScatter | | √ | | |
| FurthestPointSample | | √ | | |
| FurthestPointSampleWithDist | | √ | | |
| FusedBiasLeakyrelu | | √ | | |
| GatherPoints | | √ | | |
| GroupPoints | | √ | | |
| Iou3d | | √ | √ | |
| KNN | | √ | | |
| MaskedConv | | √ | √ | |
| MergeCells | | √ | | |
| MinAreaPolygon | | √ | | |
| ModulatedDeformConv2d | √ | √ | | |
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend (#2396) * [Feature] Support MsDeformAttnForward with cambricon MLU backend * [Feature] Support MsDeformAttnForward with cambricon MLU backend * [Feature] Support MsDeformAttnForward with cambricon MLU backend * [Feature] Support MsDeformAttnForward with cambricon MLU backend * [Feature] Support MsDeformAttnForward with cambricon MLU backend * [Feature] Support MsDeformAttnForward with cambricon MLU backend * [Feature] Support ms with cambricon MLU backend * [Feature] Support msdeformattn_1104 with cambricon MLU backend * [Feature] Support ms with cambricon MLU backend * [Feature] Support MsDeformAttn_1108 with cambricon MLU backend * [Feature] Support MsDeformAttn_1108 with cambricon MLU backend * [Feature] Support MsDeformAttn_1108 with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * Revert "[Feature] Support MsDeformAttn with cambricon MLU backend" This reverts commit 27963cccc86d240852a40a2c1510147a3e9f269f. * [Feature] Support MsdeformAttn with cambricon MLU backend * [Feature] Support MsDeformAttn with cambricon MLU backend * [Feature] Support MsdeformAttn with cambricon MLU backend Co-authored-by: zhangshaopeng <wicky-zheng@outlook.com> Co-authored-by: wicky-zheng <root@notebook-mmcv-290m8-torch1-6-y0yv66-notebook-0.notebook-mmcv-290m8-torch1-6-y0yv66.ns-ad8b689a0ecd41fcb4469c803dcd539d.svc.cluster.local>
2022-11-16 14:08:04 +08:00
| MultiScaleDeformableAttn | | √ | √ | |
| NMS | √ | √ | √ | |
| NMSRotated | √ | √ | | |
| NMSQuadri | √ | √ | | |
| PixelGroup | √ | | | |
| PointsInBoxes | √ | √ | | |
| PointsInPolygons | | √ | | |
| PSAMask | √ | √ | √ | |
| RotatedFeatureAlign | √ | √ | | |
| RoIPointPool3d | | √ | √ | |
| RoIPool | | √ | √ | |
| RoIAlignRotated | √ | √ | √ | |
| RiRoIAlignRotated | | √ | | |
| RoIAlign | √ | √ | √ | |
| RoIAwarePool3d | | √ | √ | |
| SAConv2d | | √ | | |
| SigmoidFocalLoss | | √ | √ | |
| SoftmaxFocalLoss | | √ | | |
| SoftNMS | | √ | | |
| Sparse Convolution | | √ | | |
| Synchronized BatchNorm | | √ | | |
| ThreeInterpolate | | √ | | |
| ThreeNN | | √ | √ | |
| TINShift | | √ | √ | |
| UpFirDn2d | | √ | | |
| Voxelization | √ | √ | | |
| PrRoIPool | | √ | | |