mmcv/tests/test_ops
Haodong Duan ef48a47389
[Improvement] Improve digit_version & use it for version_checking (#1185)
* improve digit_version & use it for version_checking

* more testing for digit_version

* setuptools >= 50 is needed

* fix CI

* add debuging log

* >= to ==

* fix lint

* remove

* add failure case

* replace

* fix

* consider TORCH_VERSION == 'parrots'

* add unittest

* digit_version do not deal with the case if 'parrots' in version name.
2021-07-23 21:03:33 +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 Supports cuda version BorderAlign module (#1021) 2021-05-25 15:23:11 +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 add ext ops, support parrots (#310) 2020-06-28 23:15:47 +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_deform_conv.py [Improvement] Improve digit_version & use it for version_checking (#1185) 2021-07-23 21:03:33 +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_fused_bias_leakyrelu.py [Feature]: parrots add parrots/fused_bias & upfirdn2d (#989) 2021-04-27 13:49:50 +08:00
test_info.py add ext ops, support parrots (#310) 2020-06-28 23:15:47 +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 [Improvement] Improve digit_version & use it for version_checking (#1185) 2021-07-23 21:03:33 +08:00
test_ms_deformable_attn.py Refactor the baseclass related to transformer (#978) 2021-06-11 18:09:31 +08:00
test_nms.py [Feature] NMS update (#957) 2021-05-31 23:42:26 +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_onnx.py [Feature] Add ONNX export support to torch.roll (#1194) 2021-07-20 16:54:04 +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.py fix roi_align ci for parrots (#708) 2020-12-13 20:01:25 +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_pool.py add ext ops, support parrots (#310) 2020-06-28 23:15:47 +08:00
test_saconv.py fix saconv (#489) 2020-08-16 01:45:01 +08:00
test_syncbn.py fix syncbn parameter order mismatch and parrots bug (#488) 2020-08-13 19:27:05 +08:00
test_tensorrt.py [Feature]: support tensorrt custom plugin `MMCVCornerPool` (#1179) 2021-07-09 21:07:38 +08:00
test_tensorrt_preprocess.py [Feature] NMS update (#957) 2021-05-31 23:42:26 +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