mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
c1de4c9bbf
* add ops (spconv) of mmdet3d * fix typo * refactor code * resolve comments in #1452 * fix compile error * fix bugs * fix bug * transform from 'types.h' to 'extension.h' * fix bug * transform from 'types.h' to 'extension.h' in parrots * add extension.h in pybind.cpp * add unittest * Recover code * (1) Remove prettyprint.h (2) Switch `T` to `scalar_t` (3) Remove useless lines (4) Refine example in docstring of sparse_modules.py * (1) rename from `cu.h` to `cuh` (2) remove useless files (3) move cpu files to `pytorch/cpu` * reorganize files * Add docstring for sparse_functional.py * use dispatcher * remove template * use dispatch in cuda ops * resolve Segmentation fault * remove useless files * fix lint * fix lint * fix lint * fix unittest in test_build_layers.py * add tensorview into include_dirs when compiling * recover all deleted files * fix lint and comments * recover setup.py * replace tv::GPU as tv::TorchGPU & support device guard * fix lint Co-authored-by: hdc <hudingchang.vendor@sensetime.com> Co-authored-by: grimoire <yaoqian@sensetime.com>
44 lines
740 B
Markdown
44 lines
740 B
Markdown
## CUDA 算子
|
|
|
|
MMCV 提供了检测、分割等任务中常用的 CUDA 算子
|
|
|
|
- ActiveRotatedFilter
|
|
- AssignScoreWithK
|
|
- BallQuery
|
|
- BBoxOverlaps
|
|
- CARAFE
|
|
- CrissCrossAttention
|
|
- ContextBlock
|
|
- ConvexIoU
|
|
- CornerPool
|
|
- Deformable Convolution v1/v2
|
|
- Deformable RoIPool
|
|
- DynamicScatter
|
|
- GatherPoints
|
|
- FurthestPointSample
|
|
- FurthestPointSampleWithDist
|
|
- GeneralizedAttention
|
|
- KNN
|
|
- MaskedConv
|
|
- MinAreaPolygon
|
|
- NMS
|
|
- PointsInPolygons
|
|
- PSAMask
|
|
- RotatedFeatureAlign
|
|
- RoIPointPool3d
|
|
- RoIPool
|
|
- RiRoIAlignRotated
|
|
- RoIAlign
|
|
- RoIAwarePool3d
|
|
- SimpleRoIAlign
|
|
- SigmoidFocalLoss
|
|
- SoftmaxFocalLoss
|
|
- SoftNMS
|
|
- Sparse Convolution
|
|
- Synchronized BatchNorm
|
|
- Voxelization
|
|
- ThreeInterpolate
|
|
- ThreeNN
|
|
- Weight standardization
|
|
- Correlation
|