Commit Graph

460 Commits (master)

Author SHA1 Message Date
liuyuan1-v 0d1b224fb1
[Feature] Support NmsRotated with cambricon MLU backend (#2643)
* [Feature] Support NmsRotated with cambricon MLU backend

* [Feature] remove foolproofs in nms_rotated_mlu.cpp

* [Feature] fix lint in test_nms_rotated.py

* [Feature] fix kMLU not found in nms_rotated.cpp

* [Feature] modify mlu support in nms.py

* [Feature] modify nms_rotated support in ops.md

* [Feature] modify ops/nms.py
2023-03-23 11:36:32 +08:00
duzekun 06fa32853b
[Feature] Add MLU support for Sparse Convolution op (#2589)
* [Feature] Add sparse convolution MLU API

* [Feature] update cpp code style

* end-of-file

* delete libext.a

* code style

* update ops.md

---------

Co-authored-by: budefei <budefei@cambricon.com>
2023-03-20 19:28:13 +08:00
sherie 01a0f53ea4
[Feature] Add Ascend support for RoIAlign op (#2638)
* add roi_algin op

* add unit test

* update ops.md

* fix typo

---------

Co-authored-by: momo609 <963372609.qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-20 19:10:42 +08:00
liuhw 745985937a
[Feature] Add the support of voxelization op for ascend device (#2613)
* update

* update
2023-03-03 20:07:36 +08:00
Hakjin Lee 8e8ab22686
[Enhancement] Support MultiScaleDeformableAttention with AMP (#2541)
* [Enhance] Support FP16 for MSDeformAttn

* [Fix] Data type mismatch

* Update mmcv/ops/multi_scale_deform_attn.py

* Add UT

Author:    nijkah <nijkah@gmail.com>

* Add cuda available condition

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-02-17 19:27:14 +08:00
liuhw 5a61e534ea
[Feature] Add Ascend support for bbox_overlaps (#2580)
* Add support for Ascend devices with bbox_overlaps

* ...

* ...
2023-02-06 18:33:16 +08:00
xinliang123 615d28176a
[Feature] Add Ascend support for RoIPoolGrad op (#2569)
* add roipoolgrad adapter

* amend
2023-02-06 18:29:59 +08:00
enemy1205 422816e45c
[Feature] Add C++ implementation for bbox_overlaps (#2477)
* add ops bbox_overlaps

* format code

* Return the pytorch version

* Intermediate modification

* Solve problems in parameter passing

* revise bug

* "add test case"
2023-01-31 00:04:35 +08:00
ckirchhoff 54ed0ed869
[Feature] Add support for Ascend devices with nms_rotated (#2550)
* [Feature]: add nms_rotated npu adaptater code

* [BugFix]: modify param in nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated.cpp

* [Doc]: add nms_rotated op in supported op list at ops.md

* [Test]: add nms_rotated unit_test

* [Bug]: remove device parameter in test_batched_nms function
2023-01-28 21:08:22 +08:00
liuduanhui dfb03806a1
[Feature] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520)
* [Feature] Support ball_query with cambricon MLU backend and mlu-ops library.

* [Fix] update operator data layout setting.

* [Fix] add cxx compile option to avoid symbol conflict.

* [Fix] fix lint errors.

* [Fix] update ops.md with info of ball_query support by MLU backend.

* [Feature] Fix typo.

* [Fix] Remove print.

* [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env.

* [Fix] update MMCV_MLU_OPS_PATH check logic.

* [Fix] update error info when failed to download mlu-ops.

* [Fix] check mlu-ops version matching info in mmcv.

* [Fix] revise wrong filename.

* [Fix] remove f.close and re.
2023-01-18 14:41:04 +08:00
zcc-zjut 84f60c178c
[Feature] Add support for Ascend devices with gather_points (#2555) 2023-01-18 10:33:56 +08:00
mengpenghui 71ee2a61f2
[Enhance] Add AMP support for MLU_DCNv2 (#2548) 2023-01-13 17:51:02 +08:00
mengpenghui c310d28c8f
[Feature] Add MLU support for DCN (#2540) 2023-01-13 17:46:19 +08:00
sherie 2810718a99
[Feature] Add Ascend support for RoIPool op (#2483)
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
2023-01-12 11:52:28 +08:00
ZShaopeng 48ea88ab9f
[Feature] Support Voxelization with cambricon MLU device (#2500)
* [Feature] Support hard_voxelize with cambricon MLU backend

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op
2023-01-10 19:43:45 +08:00
DaGaiBa 73856344f5
[Feature] Support PSAMask op for Ascend device (#2487) 2023-01-06 20:15:50 +08:00
Zhongyu Li 935ba78b39
[Fix] Fix skip_layer for RF-Next (#2489)
* judge skip_layer by fullname

* lint

* skip_layer first

* update unit test
2022-12-28 15:27:34 +08:00
jayggh 46eb9ec5d0
[Feature] Add NPU adapter for Deformable RoIPool (#2481)
* add npu adapter for deformroipool

* cleancode

* Update deform_roi_pool.cpp

* Update deform_roi_pool.cpp

* Update deform_roi_pool.cpp

* Update deform_roi_pool.cpp

* Update deform_roi_pool.cpp

* Update deform_roi_pool.cpp

* Update deform_roi_pool.cpp
2022-12-14 15:09:46 +08:00
jayggh 4c51afce2a
[Feature] Add FusedBiasLeakyRelu npu adapter (#2474)
* init npu

* add npu extension and focal loss adapter

* clean code

* clean code

* clean code

* clean code

* fix autocast bugs on npu (#2273)

fix autocast bugs on npu (#2273)

* code format

* code format

* code format

* bug fix

* pytorch_npu_helper.hpp clean code

* Npu dev (#2306)

* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather

* raise ImportError when compile with npu

* add npu test case (#2307)

* add npu test case

* Update focal_loss.py

* add comment

* clean lint

* update dtype assert

* update DDP forward and comment

* fix bug

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* sigmoidfocalloss npu adapter bug fix

* BugFix: modify softmaxFocalLoss adapter

* BugFix: remove equal sign in the code

* add npu install information in README

* add modulatedDeformConv npu adapter

* init npu

* add npu extension and focal loss adapter

* clean code

* clean code

* clean code

* add modulatedDeformConv npu adapter

* merge master branch 20221103

* Add masked_ Conv2d operator in NPU

* add nms_npu

* fix bug

* fix code check

* fix code check

* fix code check

* Masked_conv2d NPU

* Masked_conv2d NPU

* Masked_conv2d NPU

* remove npu-install-info in README.md

* annotate the clang-format in pre-commit-config-zh-ch.yaml

* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv

* Create fused_bias_leakyrelu_npu.cpp

Add NPU adapter for fused_bias_leaky_relu operator

* Update fused_bias_leakyrelu_npu.cpp

* Update fused_bias_leakyrelu_npu.cpp

* Update ops.md

* Update ops.md

* Update fused_bias_leakyrelu_npu.cpp

* Update fused_bias_leakyrelu_npu.cpp

* Update test_fused_bias_leakyrelu.py

* Update fused_bias_leakyrelu.py

* Update test_fused_bias_leakyrelu.py

* Update fused_bias_leakyrelu.py

* Update test_fused_bias_leakyrelu.py

* Update ops.md

* amend for CI

* bugfix

* amend ops.md

* Update test_fused_bias_leakyrelu.py

* clean code

* bugfix

* clean code

* Update fused_bias_leakyrelu_npu.cpp

* Update fused_bias_leakyrelu_npu.cpp

Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-12-13 11:13:46 +08:00
Zaida Zhou 4e67e481a0
[Fix] Fix ut caused by the new version of packaging (#2482) 2022-12-11 17:59:11 +08:00
mengpenghui 49d891754b
[Feature] Support modulated_deform_conv with cambricon MLU backend (#2411)
* [Feature] Support modulated_deform_conv with cambricon MLU backend

* fix error of torch_mlu

* modify with commit suggestion

* Update modulated_deform_conv.py

* Update mmcv/ops/modulated_deform_conv.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-04 17:46:44 +08:00
SUNYC c1712ee290
[Enhance] Add conf_path in PetrelBackend (#2459)
* feat: add conf_path in PetrelBackend

* fix: add conf_path in MockPetrelClient

* fix: adjust parameters order
2022-11-28 11:48:13 +08:00
Shanghua Gao 6e3827d02b
[Feature] Support receptive field search of CNN models (#2056)
* support rfsearch

* add labs for rfsearch

* format

* format

* add docstring and type hints

* clean code

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* rm unused func

* update code

* update code

* update code

* update  details

* fix details

* support asymmetric kernel

* support asymmetric kernel

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Apply suggestions from code review

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Apply suggestions from code review

* Apply suggestions from code review

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Apply suggestions from code review

* add unit tests for rfsearch

* set device for Conv2dRFSearchOp

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* remove unused function search_estimate_only

* move unit tests

* Update tests/test_cnn/test_rfsearch/test_operator.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update mmcv/cnn/rfsearch/operator.py

Co-authored-by: Yue Zhou <592267829@qq.com>

* change logger

* Update mmcv/cnn/rfsearch/operator.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: lzyhha <819814373@qq.com>
Co-authored-by: Zhongyu Li <44114862+lzyhha@users.noreply.github.com>
Co-authored-by: Yue Zhou <592267829@qq.com>
2022-11-22 19:15:55 +08:00
ZShaopeng 12442667ff
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend (#2396)
* [Feature] Support MsDeformAttnForward with cambricon MLU backend

* [Feature] Support MsDeformAttnForward with cambricon MLU backend

* [Feature] Support MsDeformAttnForward with cambricon MLU backend

* [Feature] Support MsDeformAttnForward with cambricon MLU backend

* [Feature] Support MsDeformAttnForward with cambricon MLU backend

* [Feature] Support MsDeformAttnForward with cambricon MLU backend

* [Feature] Support ms with cambricon MLU backend

* [Feature] Support msdeformattn_1104 with cambricon MLU backend

* [Feature] Support ms with cambricon MLU backend

* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend

* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend

* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* Revert "[Feature] Support MsDeformAttn with cambricon MLU backend"

This reverts commit 27963cccc86d240852a40a2c1510147a3e9f269f.

* [Feature] Support MsdeformAttn with cambricon MLU backend

* [Feature] Support MsDeformAttn with cambricon MLU backend

* [Feature] Support MsdeformAttn with cambricon MLU backend

Co-authored-by: zhangshaopeng <wicky-zheng@outlook.com>
Co-authored-by: wicky-zheng <root@notebook-mmcv-290m8-torch1-6-y0yv66-notebook-0.notebook-mmcv-290m8-torch1-6-y0yv66.ns-ad8b689a0ecd41fcb4469c803dcd539d.svc.cluster.local>
2022-11-16 14:08:04 +08:00
ckirchhoff de6b0021af
[Feature] Support masked_conv in Ascend device (#2387)
* init npu

* add npu extension and focal loss adapter

* clean code

* clean code

* clean code

* clean code

* fix autocast bugs on npu (#2273)

fix autocast bugs on npu (#2273)

* code format

* code format

* code format

* bug fix

* pytorch_npu_helper.hpp clean code

* Npu dev (#2306)

* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather

* raise ImportError when compile with npu

* add npu test case (#2307)

* add npu test case

* Update focal_loss.py

* add comment

* clean lint

* update dtype assert

* update DDP forward and comment

* fix bug

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* sigmoidfocalloss npu adapter bug fix

* BugFix: modify softmaxFocalLoss adapter

* BugFix: remove equal sign in the code

* add npu install information in README

* add modulatedDeformConv npu adapter

* init npu

* add npu extension and focal loss adapter

* clean code

* clean code

* clean code

* add modulatedDeformConv npu adapter

* merge master branch 20221103

* Add masked_ Conv2d operator in NPU

* add nms_npu

* fix bug

* fix code check

* fix code check

* fix code check

* Masked_conv2d NPU

* Masked_conv2d NPU

* Masked_conv2d NPU

* remove npu-install-info in README.md

* annotate the clang-format in pre-commit-config-zh-ch.yaml

* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv

Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-11-15 10:23:17 +08:00
q.yao 6254ebef8d
[Fix] Create Tensor with new_* method to support amp (#2389) 2022-11-08 15:54:34 +08:00
Zhang b622fb2e29
[Feature] Support RoIAwarePool3d with cambricon MLU backend (#2359)
* [Feature] Support RoiawarePool3d with cambricon MLU backend.

* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-03 11:13:34 +08:00
VVsssssk 93fe4829f7 [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 16:53:33 +08:00
Yue Zhou a4f304a5f5 [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-28 16:53:33 +08:00
takuoko 9709ff3f8c
[Enhancement] Add a new argument define_metric in wandb hook (#2237)
* wandb define_metric

* add test and some fix based on mmengine PR

* fix test

* add summary warnings

* Update mmcv/runner/hooks/logger/wandb.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update mmcv/runner/hooks/logger/wandb.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-24 23:13:07 +08:00
cathyzhang222 7fd7058a9c
[Feature] Support Iou3d with cambricon MLU backend (#2339)
* [Feature] Support Iou3d with cambricon MLU backend

* [Feature] add double line

* [Feature] add line
2022-10-24 15:58:58 +08:00
q.yao 6cf9ac0507
[Fix] Fix three nn op can not accept half tensor (#2348)
* Fix three nn half inpt

* update test
2022-10-24 11:15:21 +08:00
Ivan Shcheklein e417035f5d
[Enhancement] Add ability to pass logger instance to frameworks (#2317)
* Add ability to pass logger instance to frameworks

* refine docstring

* Update mmcv/runner/hooks/logger/dvclive.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-12 12:09:53 +08:00
CokeDong dd5b415d2a
[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-11 15:21:00 +08:00
q.yao c001e2fcba
[Refactor] Refactor the nms3d op to support MLU (#2296)
* update iou3d

* fix parrot

* update to device

* replace count_nonzero with fill

* update build.yml
2022-10-11 10:24:46 +08:00
Zaida Zhou b6a7fd98e4
Upgrade pre commit hooks (#2321)
* Upgrade the versions of pre-commit hooks

* update the versions of zh-cn.yaml
2022-10-08 11:48:44 +08:00
wangjiangben-hw a5db5f6682
[Feature] Support training on NPU device (#2262)
* init npu

* add npu extension and focal loss adapter

* clean code

* clean code

* clean code

* clean code

* fix autocast bugs on npu (#2273)

fix autocast bugs on npu (#2273)

* code format

* code format

* code format

* bug fix

* pytorch_npu_helper.hpp clean code

* Npu dev (#2306)

* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather

* raise ImportError when compile with npu

* add npu test case (#2307)

* add npu test case

* Update focal_loss.py

* add comment

* clean lint

* update dtype assert

* update DDP forward and comment

* fix bug

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-09-30 21:05:37 +08:00
takuoko afff388692
[Enhancement] Support SiLU with torch < 1.7.0 (#2278)
* support silu torch<1.7.0

* fix test

* fix test

* fix inplace

* fix inplace

* Update tests/test_cnn/test_silu.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-09-29 15:19:41 +08:00
takuoko 55c51e1734
[Feature] Register a new activatation layer SiLU to ACTIVATION_LAYERS (#2256) 2022-09-13 22:38:23 +08:00
ZShaopeng 83bedf5a2b
[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-09-08 17:19:12 +08:00
bdf e843d73fc7
[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-08-28 22:37:28 +08:00
liuduanhui a08517790d
[Feature] Support ThreeNN with cambricon MLU backend (#2215) 2022-08-23 15:18:47 +08:00
bdf e663670a74
[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-21 23:21:54 +08:00
bdf 832b845257
[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-19 10:36:26 +08:00
Jingwei Zhang b9bdfeb6f6
[Enhance] Support different border modes in imrotate (#2141)
* support different border_mode in imrotate

* fix docstring

* small fix

* reorder the border_mode arg in imrotate

* fix lint

* polish docstring

* polish docstring
2022-08-17 10:53:05 +08:00
Yosuke Shinya 304f184c46
[Fix] Fix gradient cumulative optimizer when resuming (#2093)
* Fix gradient cumulative optimizer when resuming

* add unit test
2022-08-15 10:50:43 +08:00
Yixiao Fang b2ac245602 [Refactor] Add pillow backend for ColorJitter related functions (#2127)
* add 'backend' for ColorJitter related functions

* add unittest

* fix unittest

* add bgr transpose and revise ut

* update unittest

* revise docstring
2022-08-03 16:21:41 +08:00
xcnick 4558bfbd5b
[Fix] Fix onnx unit tests (#2155) 2022-07-29 19:23:29 +08:00
Peter Ye 78f01001d5
[Feature] Support setting start epoch and interval epoch for PAVI Logger Hook (#2103)
* [Feature] Support setting start epoch and interval epoch for PAVI Logger Hook

* [Feature] Update the coding style as the maintainer wish

* fix: default integer division or modulo by zero

* fix: runner.epoch is less than start and use self.get_epoch instead of runner.epoch

* feat: support for iter-based runner and fix the step bug

* feat: iter based hook

* feat: fix bug and coding style

* fix: coding style

* fix: coding style

* fix: graph may add in evaluation
2022-07-29 11:40:20 +08:00
Jingwei Zhang 35affc35bf
[Feature] Support PrRoIPool operation (#2113)
* support prroipooling

* fix build bug

* fix bug of precision

* fix lint

* add ut

* add license and citation

* polish op and add type hint

* refactor ut

* add docstring for prroipool

* polish code

* format cu and cuh code

* fix format

* fix format

* fix typo

Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-07-23 23:08:32 +08:00