mmcv/tests/test_cnn
Wenhao Wu c1de4c9bbf
[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) 2022-02-12 14:34:35 +08:00
test_depthwise_seperable_conv_module.py Add Depthwise Seperable ConvModule (#477) 2020-08-09 17:29:02 +08:00
test_flops_counter.py [Feature] support to calculate FLOPs of GN, IN, LN (#897) 2021-03-19 14:04:49 +08:00
test_fuse_conv_bn.py Move fuse conv bn to mmcv (#382) 2020-07-08 19:03:15 +08:00
test_generalized_attention.py [Fix] fix generalized attention fp16 (#1036) 2021-05-23 15:16:27 +08:00
test_hsigmoid.py [Fix] Align HSigmoid with pytorch official implementation(#1622) 2022-01-20 14:25:10 +08:00
test_hswish.py
test_model_registry.py add model registry (#760) 2021-04-10 21:33:34 +08:00
test_non_local.py Add pairwise function for 'gaussian' and 'concatenation' mode in NonLocal. (#424) 2020-07-17 23:44:50 +08:00
test_revert_syncbn.py [Feature] Add revert_sync_batchnorm (#1253) 2021-09-08 10:59:39 +08:00
test_scale.py
test_swish.py [Fix] Fix some warnings in unittest (#1522) 2021-12-22 10:57:10 +08:00
test_transformer.py [Fix] Fix an incorrect assignment in the BaseTransformerLayer (#1670) 2022-01-28 23:43:53 +08:00
test_weight_init.py [Docs] Add header for files (#1181) 2021-08-08 22:10:55 +08:00
test_wrappers.py empty tensor inference backward compatible (#1131) 2021-06-24 17:04:03 +08:00