Zaida Zhou 97e5bada4c

continue PR #1223 (#1404) ...

* fix MultiScaleDeformableAttention inference issue on cpu model

* fix lint

* add unintest

* remove some code

* Update tests/test_ops/test_ms_deformable_attn.py

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

* fix device

* remove device

* add more device

* refactor unittest

Co-authored-by: zhicheng huang <zhichenghzc@gmail.com>
Co-authored-by: zhangshilong <2392587229zsl@gmail.com>
Co-authored-by: Shilong Zhang <61961338+jshilong@users.noreply.github.com>

2021-10-14 20:50:38 +08:00

..

test_ball_query.py

[Feature] Add ballquery op from mmdet3d (#1332)

2021-09-23 21:08:00 +08:00

test_bbox.py

Misc updates regarding Windows and unittests (#586)

2020-10-28 16:23:19 +08:00

test_bilinear_grid_sample.py

[Feature] enable exporting to onnx for PointRend (#953)

2021-06-11 13:49:19 +08:00

test_border_align.py

Add code spell hook to pre-commit and fix typos (#1384)

2021-10-04 20:13:54 +08:00

test_box_iou_rotated.py

add 'iof' mode for box_iou_rotated (#753)

2020-12-25 17:06:45 +08:00

test_carafe.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_cc_attention.py

Reimplement cc_attention using pure pytorch (#1201)

2021-09-09 21:53:45 +08:00

test_contour_expand.py

Add pixel group and contour expand ops (#993)

2021-05-12 10:41:22 +08:00

test_corner_pool.py

Migrate op (#392)

2020-07-08 17:29:15 +08:00

test_correlation.py

[Fix] Revise unit test of correlation (#1368)

2021-09-25 21:13:34 +08:00

test_deform_conv.py

Add DCN and Modulated DCN CPU implementation (#1278)

2021-08-29 20:48:31 +08:00

test_deform_roi_pool.py

Add new parrots extension implementation for all ops (#794)

2021-02-26 19:05:25 +08:00

test_focal_loss.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_furthest_point_sample.py

[Feature] Add fps op from mmdet3d (#1337)

2021-10-13 20:39:31 +08:00

test_fused_bias_leakyrelu.py

[Feature]: parrots add parrots/fused_bias & upfirdn2d (#989)

2021-04-27 13:49:50 +08:00

test_gather_points.py

[Feature] Add gather points op from mmdet3d (#1338)

2021-10-14 16:58:12 +08:00

test_info.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_knn.py

[Feature] Add knn op from mmdet3d (#1354)

2021-10-14 13:33:13 +08:00

test_masked_conv2d.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_merge_cells.py

Migrate op (#392)

2020-07-08 17:29:15 +08:00

test_modulated_deform_conv.py

Add DCN and Modulated DCN CPU implementation (#1278)

2021-08-29 20:48:31 +08:00

test_ms_deformable_attn.py

continue PR #1223 (#1404)

2021-10-14 20:50:38 +08:00

test_nms_rotated.py

Fix some bug as well as unit test for nms_rotate (#728)

2020-12-18 15:17:11 +08:00

test_nms.py

[Feature] NMS update (#957)

2021-05-31 23:42:26 +08:00

test_onnx.py

Add Deformable Conv CustomOp for onnxruntime (#1343)

2021-10-13 20:28:24 +08:00

test_pixel_group.py

Add pixel group and contour expand ops (#993)

2021-05-12 10:41:22 +08:00

test_psa_mask.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_roi_align_rotated.py

[Feature]: Add Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda) and onnxruntime (cpu) (#933)

2021-04-19 15:04:01 +08:00

test_roi_align.py

fix roi_align ci for parrots (#708)

2020-12-13 20:01:25 +08:00

test_roi_pool.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_saconv.py

Add DCN and Modulated DCN CPU implementation (#1278)

2021-08-29 20:48:31 +08:00

test_syncbn.py

[Feature]: Support empty tensor in MMSyncBN (#1205)

2021-09-23 17:10:38 +08:00

test_tensorrt_preprocess.py

[Feature] NMS update (#957)

2021-05-31 23:42:26 +08:00

test_tensorrt.py

Add code spell hook to pre-commit and fix typos (#1384)

2021-10-04 20:13:54 +08:00

test_tin_shift.py

Add tin_shift function (#492)

2020-08-15 22:29:07 +08:00

test_upfirdn2d.py

[Feature]: parrots add parrots/fused_bias & upfirdn2d (#989)

2021-04-27 13:49:50 +08:00