mmcv/test_cnn at c1de4c9bbf427b3786546eb16caf37f29cc39651 - mmcv

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

History

[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

test_build_layers.py

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

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

test_context_block.py

Refactor plugins: move from ops to cnn (#380 )

2020-07-03 01:37:58 +08:00

test_conv2d_adaptive_padding.py

Implementation of 2D convolution in tensorflow with padding as "same" (#529 )

2020-09-09 23:57:48 +08:00

test_conv_module.py

[Enhance] Use PyTorch HSwish implementation. (#1709 )