7156604eba
[Feature] Add BezierAlign CUDA op ( #2393 )
...
* bezier align
* add ut
* fix comment
* updata ut
* fix link and comment
* fix comment
2023-01-10 17:05:39 +08:00
433607030a
[Enhancement] Support value_proj_ratio in MultiScaleDeformableAttention ( #2452 )
...
* add ratio in ms_deform_attn_
* add ratio in ms_deform_attn
* Update mmcv/ops/multi_scale_deform_attn.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_ops/test_ms_deformable_attn.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add ratio in ms_deform_attn
* add ratio in ms_deform_attn
* add ratio in ms_deform_attn
* add ratio in ms_deform_attn
* add ratio in ms_deform_attn
* add ratio in ms_deform_attn
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-11 17:48:21 +08:00
a0939977a3
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend
2022-11-23 16:57:35 +08:00
6c89b71756
[Feature] Support Iou3d with cambricon MLU backend ( #2339 )
...
* [Feature] Support Iou3d with cambricon MLU backend
* [Feature] add double line
* [Feature] add line
2022-11-23 16:57:35 +08:00
e9777adbac
[Feature] Support RoIAwarePool3d with cambricon MLU backend ( #2359 )
...
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-23 16:57:35 +08:00
2d10616b8f
[Fix] Fix three nn op can not accept half tensor ( #2348 )
...
* Fix three nn half inpt
* update test
2022-11-23 16:57:35 +08:00
3bb0611aef
[Fix] Create Tensor with new_* method to support amp ( #2389 )
2022-11-23 16:57:35 +08:00
4e0341f26f
[Features] Add stack ball query and stack group points ops ( #2292 )
...
* add stack sa model ops
* fix lint
* fix lint
* fix comments
* fix bug
* fix lint
* fix comments
* fix lint
* fix lint
* fix
2022-10-28 00:42:10 +08:00
a8f7ae48e2
[Feature] Support RoipointPool3d with cambricon MLU backend ( #2247 )
...
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
2022-10-22 13:39:50 +08:00
a364e6cad2
[Feature] Support DeformRoiPool with cambricon MLU backend ( #2137 )
...
* [Feature] Support DeformRoiPool with cambricon MLU backend
* [Fix] Remove use of std library
* [Fix] Correct the error information
* [Refactor] Refactor test deform_roi_pool code
* [Fix] Fix judgment error
* [Fix] Modify the large tensor check
Co-authored-by: budefei <budefei@cambricon.com>
2022-10-22 13:39:50 +08:00
8fae3b4682
[Feature] Support PrRoI op for Parrots ( #2280 )
...
* Support parrots extension for op PrRoI
* Fix lint
* Fix cpp lint
* Fix testcase failure by false requires_grad in self-defined autograd Funtion
* Fix issues
* Fix flake8
* Fix isort
* Adaption for typechecking for PrRoIPoolFunction
* Fix lint
* Support only float32
* bugfix
2022-10-22 13:39:50 +08:00
e0b3223baf
fix ci
2022-10-22 13:39:50 +08:00
cb2eb57683
Remove onnx ( #2295 )
...
* remove onnx
* remove print
2022-10-21 15:35:07 +08:00
a4c82617ff
[Feature] Add box_iou_quadri & nms_quadri ( #2277 )
...
* update
* update
* fix link
* fix bug
* update nms_quadri
* fix lint
* Update test_nms_quadri.py
* Update box_iou_quadri.py
* fix bug
* Update test_nms_quadri.py
* Update box_iou_rotated_utils.hpp
* Update box_iou_quadri.py
* Update mmcv/ops/nms.py
2022-10-13 17:26:14 +08:00
cdfbdc0bd1
[Feature] Support ThreeNN with cambricon MLU backend ( #2215 )
2022-08-28 17:54:20 +08:00
625e82ce62
[Feature] Add carafe op for MLU ( #2212 )
...
* [Feature] Support CARAFE with Cambricon MLU backend
* [Docs] Add comments for common funtions
* [Test] Add allclose test for carafe
* Remove print
Co-authored-by: zcyKTH <zcy19950525@gmail.com>
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 17:54:20 +08:00
f5a19ef06f
[Feature] Support MaskedConv2d with cambricon MLU backend ( #2202 )
...
* [Feature] Support MaskedConv2d with cambricon MLU backend
* [Refactor] Refactor test masked_conv2d code
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 17:54:20 +08:00
2e5628b410
[Refactor]: Remove deployment for dev-2.x ( #2225 )
...
* remove deploy for 2.0
* update onnx ut
2022-08-26 20:11:05 +08:00
c3835415ed
[Refactor] Modify the importing path due to mmengine changes ( #2224 )
2022-08-24 19:48:10 +08:00
5947178e85
Remove many functions in utils and migrate them to mmengine ( #2217 )
...
* Remove runner, parallel, engine and device
* fix format
* remove outdated docs
* migrate many functions to mmengine
* remove sync_bn.py
2022-08-23 10:12:41 +08:00
0fe1c64744
Remove fileio from mmcv and use mmengine.fileio instead ( #2179 )
2022-08-16 13:37:19 +08:00
0b4285d959
Pick commits from master ( #2164 )
...
* [Docs] Add swith_language.md in docs (#2160 )
* [Fix] Fix onnx unit tests (#2155 )
* [Docs] Limit extension versions (#2144 )
* Support PrRoIPool operation
* Add MPS bbox overlap
* Add .pre-commit-config-zh-cn.yaml (#2135 )
Co-authored-by: xcnick <xcnick0412@gmail.com>
Co-authored-by: Jingwei Zhang <zjw18@mails.tsinghua.edu.cn>
Co-authored-by: q.yao <yaoqian@sensetime.com>
2022-08-03 14:11:53 +08:00
2708fac6c5
[Feature] Support RoiPool with Cambricon MLU backend ( #2073 )
...
* [Feature] Support RoiPool with cambricon MLU backend
* [Docs] Update ops.md
2022-06-29 11:24:00 +08:00
9807c2d268
[Fix] Fix batched_nms for rotated box and add type hints for nms.py ( #2006 )
...
* Fix batched_nms for rotated box.
Add type hint for nms.py
* Add test
* doc string
* revert symbolic hint
* fix max_coordinate
* add comment
* rename type
* fix typo docstring
2022-06-10 19:49:07 +08:00
15495ea0a7
[Fix] make iou3d.boxes_iou3d actually calculate 3D IoU ( #2018 )
...
* fix iou3d
* rename variables in test
* fix comments
2022-06-09 21:08:32 +08:00
834d5978df
[Feature] Add ChamferDistance op in gpu ( #1933 )
...
* update
* fix lint
* fix lint
* Update mmcv/ops/chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update chamfer_distance.py
* Update test_chamfer_distance.py
* fix
* Update chamfer_distance_cuda_kernel.cuh
* Update chamfer_distance_cuda_kernel.cuh
* fix
* Update mmcv/ops/chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update chamfer_distance.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-08 18:12:57 +08:00
5601427bbc
[Feature] Support PSAMask with cambricon MLU backend ( #2024 )
...
* [Feature] Support Psamask with cambricon MLU backend
* [Fix] refine the format according to lint check
* Replace std:max/min with conditional operators
* [Fix] Format head files
2022-06-08 16:13:13 +08:00
fb5062ca00
[Feature] Support RoiAlignRotated with cambricon MLU backend ( #2033 )
...
* [Feature] Support RoiAlignRotated with cambricon MLU backend
* [Fix] Remove std lib in mlu files
* [Fix] replace std::min/max with conditional operators
2022-06-08 14:13:28 +08:00
e9f48a4f8e
[Enhance] Replace BEV IoU with 3D IoU ( #1902 )
...
* add iou3d
* revert deprecated python function
* fix lint
* replace 3d iou/nms calls for bev iou/nms
2022-05-25 17:05:50 +08:00
7e6f462495
[Fix] Skip CPU test in test_rotated_feature_align.py for parrots ( #1968 )
...
* skip test_rotated_feature_align.py for parrots
Signed-off-by: del-zhenwu <dele.zhenwu@gmail.com>
* fix lint
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-05-20 17:04:58 +08:00
60eadb0665
[Enhancement] Lower the restrictions of _resize method in BaseMergeCell ( #1959 )
...
* Fix the bug met in using nasfpn
Fix the bug met in using nasfpn which is mentioned at https://github.com/open-mmlab/mmdetection/issues/5987 .
Avoid the strong restrictions of _resize function in BaseMergeCell:
1. When Downsampling the feature map, the feature map's shape must be divisible by the target size. We pad zero around feature map before max_pool2d opt to make it always divisible. (line 102 ~ 107)
2. Considering the different downsampling scale of H and W, shape[-2] and shape[-1] are involed in the definition of kernel_size. (line 110)
* Update merge_cells.py
check flake8 & isort
* Update merge_cells.py
* Update merge_cells.py
yapf
* Update mmcv/ops/merge_cells.py
X_pad rename to padding_x
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
* Update merge_cells.py
format the code style after renameing the X_pad to padding_x
* Update test_merge_cells.py
Mainly test the downsampling resize in BaseMergeCell. The smaller target size is set to (14, 7), the classical feature map's size in the last few stages of the backbone, which will product different downsampling scales in different dims.
* Update test_merge_cells.py
add "# Copyright (c) OpenMMLab. All rights reserved."
* Update merge_cells.py
format the variable name
* Update test_merge_cells.py
Testing divisible and indivisible situations simultaneously
* Update mmcv/ops/merge_cells.py
fix the bug when h is indivisible and w is divisible, the pad_w will be padded unreasonable.
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/merge_cells.py
fix the bug when w is indivisible and h is divisible, the pad_h will be padded unreasonable.
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix undefined error
* Update merge_cells.py
make pad_h, pad_w more readable
* Update test_merge_cells.py
use @pytest.mark.parametrize instead of 'for' methor
* Update merge_cells.py
* Update test_merge_cells.py
isort
* Update merge_cells.py
isort
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-20 15:13:13 +08:00
45fa3e44a2
Add pyupgrade pre-commit hook ( #1937 )
...
* add pyupgrade
* add options for pyupgrade
* minor refinement
2022-05-18 11:47:14 +08:00
e70a318ae9
[Fix] Fix parrots unit tests ( #1913 )
2022-05-09 16:59:32 +08:00
b30755eef2
[Feature] Add rotated_feature_align cpu & onnxruntime implementation ( #1878 )
...
* add rotated_feature_align cpu implementation
* add rotated_feature_align onnxruntime implementation
* Update code for advices from grimoire
Remove useless comment from mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
Replace ambiguous function name atomicAdd in mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
Simplify unit test with parameter in tests/test_ops/test_rotated_feature_align.py
Use fma in interpolate in mmcv/ops/csrc/onnxruntime/cpu/rotated_feature_align.cpp mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
* Inline the function to reduce the overhead of the function call
Use fma in interpolate
2022-04-29 10:02:33 +08:00
de0c1039f7
Fix bbox unit test ( #1914 )
2022-04-26 16:49:28 +08:00
0e59036211
[Enhancement] Optimize bbox overlap ( #1718 )
...
* add half support
* add cpu implementation
* fix bugs, load with inline asm
* better vector load
* add comments
2022-04-26 00:44:01 +08:00
03b38a46ad
[Refactor] Move is_mlu_available to mmcv/utils/device_type.py ( #1884 )
...
* [Refactor] Move is_mlu_available to mmcv/utils/device_type.py
* remove comma
* fix isort
2022-04-17 00:15:35 +08:00
362a90f8bf
[Feature] Add several MLU ops ( #1563 )
...
* [Feature] Add roiaware pool3d ops from mmdet3d (#1382 )
* add ops (roiaware pool3d) in mmdet3d
* refactor code
* fix typo
Co-authored-by: zhouzaida <zhouzaida@163.com>
* [Feature] Add iou3d op from mmdet3d (#1356 )
* add ops (iou3d) in mmdet3d
* add unit test
* refactor code
* refactor code
* refactor code
* refactor code
* refactor code
Co-authored-by: zhouzaida <zhouzaida@163.com>
* [Fix] Update test data for test_iou3d (#1427 )
* Update test data for test_iou3d
* delete blank lines
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* [Feature] Add group points ops from mmdet3d (#1415 )
* add op (group points) and its related ops (ball query and knn) in mmdet3d
* refactor code
* fix typo
* refactor code
* fix typo
* refactor code
* make input contiguous
Co-authored-by: zhouzaida <zhouzaida@163.com>
* add mmdet3d op (#1425 )
Co-authored-by: zhouzaida <zhouzaida@163.com>
* [Feature] Loading objects from different backends and dumping objects to different backends (#1330 )
* [Feature] Choose storage backend by the prefix of filepath
* refactor FileClient and add unittest
* support loading from different backends
* polish docstring
* fix unittet
* rename attribute str_like_obj to is_str_like_obj
* add infer_client method
* add check_exist method
* rename var client to file_client
* polish docstring
* add join_paths method
* remove join_paths and add _format_path
* enhance unittest
* refactor unittest
* singleton pattern
* fix test_clientio.py
* deprecate CephBackend
* enhance docstring
* refactor unittest for petrel
* refactor unittest for disk backend
* update io.md
* add concat_paths method
* improve docstring
* improve docstring
* add isdir and copyfile for file backend
* delete copyfile and add get_local_path
* remove isdir method of petrel
* fix typo
* add comment and polish docstring
* polish docstring
* rename _path_mapping to _map_path
* polish docstring and fix typo
* refactor get_local_path
* add list_dir_or_file for FileClient
* add list_dir_or_file for PetrelBackend
* fix windows ci
* Add return docstring
* polish docstring
* fix typo
* fix typo
* deprecate the conversion from Path to str
* add docs for loading checkpoints with FileClient
* refactor map_path
* add _ensure_methods to ensure methods have been implemented
* fix list_dir_or_file
* rename _ensure_method_implemented to has_method
* Add CI for pytorch 1.10 (#1431 )
* [Feature] Upload checkpoints and logs to ceph (#1375 )
* [Feature] Choose storage backend by the prefix of filepath
* refactor FileClient and add unittest
* support loading from different backends
* polish docstring
* fix unittet
* rename attribute str_like_obj to is_str_like_obj
* [Docs] Upload checkpoint to petrel oss
* add infer_client method
* Support uploading checkpoint to petrel oss
* add check_exist method
* refactor CheckpointHook
* support uploading logs to ceph
* rename var client to file_client
* polish docstring
* enhance load_from_ceph
* refactor load_from_ceph
* refactor TextLoggerHook
* change the meaning of out_dir argument
* fix test_checkpoint_hook.py
* add join_paths method
* remove join_paths and add _format_path
* enhance unittest
* refactor unittest
* add a unittest for EvalHook when file backend is petrel
* singleton pattern
* fix test_clientio.py
* deprecate CephBackend
* add warning in load_from_ceph
* fix type of out_suffix
* enhance docstring
* refactor unittest for petrel
* refactor unittest for disk backend
* update io.md
* add concat_paths method
* fix CI
* mock check_exist
* improve docstring
* improve docstring
* improve docstring
* improve docstring
* add isdir and copyfile for file backend
* delete copyfile and add get_local_path
* remove isdir method of petrel
* fix typo
* rename check_exists to exists
* refactor code and polish docstring
* fix windows ci
* add comment and polish docstring
* polish docstring
* polish docstring
* rename _path_mapping to _map_path
* polish docstring and fix typo
* refactor get_local_path
* add list_dir_or_file for FileClient
* add list_dir_or_file for PetrelBackend
* fix windows ci
* Add return docstring
* polish docstring
* fix typo
* fix typo
* fix typo
* fix error when mocking PetrelBackend
* deprecate the conversion from Path to str
* add docs for loading checkpoints with FileClient
* rename keep_log to keep_local
* refactor map_path
* add _ensure_methods to ensure methods have been implemented
* fix list_dir_or_file
* rename _ensure_method_implemented to has_method
* refactor
* polish information
* format information
* bump version to v1.3.16 (#1430 )
* [Fix]: Update test data of test_tin_shift (#1426 )
* Update test data of test_tin_shift
* Delete tmp.engine
* add pytest raises asserterror test
* raise valueerror, update test log
* add more comment
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix the wrong function reference bug in BaseTransformerLayer when batch_first is True (#1418 )
* [Docs] Add mmcv itself in the docs list (#1441 )
* Add mmcv itself in the docs list
* modify link of docs
* [Improve] improve checkpoint loading log (#1446 )
* [Feature] Support SigmoidFocalLoss with Cambricon MLU backend (#1346 )
* [Feature] Support SigmoidFocalLoss with Cambricon MLU backend
* refactor MMCV_WITH_MLU macro define
* refactor NFU_ALIGN_SIZE, PAD_DOWN and split_pipeline_num
* delete extra fool proofing in cpp
* [Feature] Support SigmoidFocalLossBackward with Cambricon MLU backend
* fix macro definition in SigmoidFocalLoss
* refactor mlu files into clang-format
* refactor sigmoid focal loss test
* refactor Sigmoid Focal Loss file structure.
* fix python lint error
* fix import torch_mlu error type
* fix lint
* refactor clang format style to google
Co-authored-by: zhouzaida <zhouzaida@163.com>
* [Feature] Support RoiAlign With Cambricon MLU Backend (#1429 )
* [Feature] Support NMS with cambricon MLU backend (#1467 )
* [Feature] Support BBoxOverlaps with cambricon MLU backend (#1507 )
* [Refactor] Format C++ code
* [Refactor] include common_mlu_helper in pytorch_mlu_helper and refactor build condition
* [Improve] Improve the performance of roialign, nms and focalloss with MLU backend (#1572 )
* [Improve] Improve the performance of roialign with MLU backend
* replace CHECK_MLU with CHECK_MLU_INPUT
* [Improve] Improve the perf of nms and focallosssigmoid with MLU backend
* [Improve] Improve the performance of roialign with MLU backend (#1741 )
* [Feature] Support tin_shift with cambricon MLU backend (#1696 )
* [Feature] Support tin_shift with cambricon MLU backend
* [fix] Add the assertion of batch_size in tin_shift.py
* [fix] fix the param check of tin_shift in cambricon code
* [fix] Fix lint failure.
* [fix] Fix source file lint failure.
* Update mmcv/ops/tin_shift.py
[Refactor] Modify the code in mmcv/ops/tin_shift.py.
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: budefei <budefei@cambricon.com>
Co-authored-by: budefei <budefei@cambricom.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* resolve conflicts and fix lint
* fix mmcv.utils.__init__
* fix mmcv.utils.__init__
* Fix lints and change FLAG
* fix setup and refine
* remove a redundant line
* remove an unnecessary 'f'
* fix compilation error
Co-authored-by: dingchang <hudingchang.vendor@sensetime.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: q.yao <yaoqian@sensetime.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: pc <luopeichao@sensetime.com>
Co-authored-by: Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
Co-authored-by: q.yao <streetyao@live.com>
Co-authored-by: Tong Gao <gaotongxiao@gmail.com>
Co-authored-by: Yuxin Liu <liuyuxin@cambricon.com>
Co-authored-by: zihanchang11 <92860914+zihanchang11@users.noreply.github.com>
Co-authored-by: shlrao <shenglong.rao@gmail.com>
Co-authored-by: zhouchenyang <zcy19950525@gmail.com>
Co-authored-by: Mrxiaofei <36697723+Mrxiaofei@users.noreply.github.com>
Co-authored-by: budefei <budefei@cambricon.com>
Co-authored-by: budefei <budefei@cambricom.com>
2022-04-16 15:45:00 +08:00
aee596d523
[Enhance] Differentiable rotated IoU ( #1854 )
...
* diff_iou_rotated is working
* add test; fix lint
* fix lint for test
* disable cpu build
* refactor files structure
* fix comments
* remove extra .repeat()
* add comment
* fix j-1 bug; update doc
* fix clang lint
* update docstrings
* fix comments
* fix comments
2022-04-15 15:40:07 +08:00
51b7a6d74a
[Fix] Fix spconv unit test by releasing cache on GPU ( #1872 )
2022-04-14 22:35:29 +08:00
6289b6f922
[Fix] Fix rocm support ( #1704 )
2022-03-21 15:24:22 +08:00
68a2c0a1e2
Update unittest for mmdet3d ops ( #1811 )
2022-03-18 12:52:55 +08:00
b5d550f090
[Feature] Add nondeterministic voxelization op from mmdet3d ( #1783 )
...
* add nondeterministic voxelization op
* fix lint
* fix lint
* resolve comments
* fix lint
2022-03-15 14:21:34 +08:00
09b64a60b0
[Fix] Fix bugs in DynamicScatter op ( #1748 )
...
* Fix bugs in DynamicScatter op
* recover unittest
* add a comment as a reminder
* compatible to torch with lower version
2022-03-07 21:02:38 +08:00
6e9ce18323
Add copyright pre-commit-hook ( #1742 )
...
* first commit
* Add copyright pre-commit-hook
2022-02-24 09:24:25 +08:00
e7f049afa1
Add mmrotate link in README.md ( #1733 )
...
* Add mmrotate link in README.md
* Add mmrotate link in docs
* fix isort
2022-02-20 11:42:33 +08:00
b83bdb0cbe
[Refactor] Refactor the interface for RoIAlignRotated ( #1662 )
...
* fix interface for RoIAlignRotated
* Add a unit test for RoIAlignRotated
* Make a unit test for RoIAlignRotated concise
* fix interface for RoIAlignRotated
* Refactor ext_module.nms_rotated
* Lint cpp files
2022-02-18 16:59:33 +08:00
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
b10eaedd38
[Fix] batched nms for nms_rotated ( #1688 )
...
* Fix batched_nms to support nms_rotated
* Update test code for batched_nms with nms_rotated
* Edit indexing the score in nms
* Use -1 indexing for batched_nms
2022-02-12 14:45:39 +08:00
dd82ab1497
Upgrade the version of isort ( #1705 )
...
* Upgrade the version of isort
* sort the order of importing modules
2022-02-09 21:29:21 +08:00
liukuikun
takuoko
ZShaopeng
cathyzhang222
Zhang
q.yao
VVsssssk
bdf
CokeDong
Yue Zhou
liuduanhui
Zaida Zhou
zhouchenyang
Yanyi Liu
Danila Rukhovich
del-zhenwu
JarvisKevin
pc
WilliamKyle
q.yao
Jiazhen Wang
Wenhao Wu
Hakjin Lee