mmcv/ops.md at 09b64a60b04a3dae5fed6ec7b557c5edca5b5cc0

mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00

[Feature] Add spconv ops from mmdet3d (#1581 )

* 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>

2022-02-18 15:53:10 +08:00

740 B

Raw Blame History

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

740 B Raw Blame History

CUDA 算子

740 B

Raw Blame History