diff --git a/mlu_op_test.proto b/mlu_op_test.proto index abc16f7..58926cb 100755 --- a/mlu_op_test.proto +++ b/mlu_op_test.proto @@ -229,6 +229,93 @@ message SeqData { optional RandomData random_data = 14; } +enum OpType { + ABS = 100; + ACTIVE_ROTATED_FILTER_FORWARD = 101; + ADDN = 102; + BALL_QUERY = 200; + BBOX_OVERLAPS = 201; + BORDER_ALIGN_BACKWARD = 202; + BORDER_ALIGN_FORWARD = 203; + BOX_IOU_ROTATED = 204; + CARAFE_BACKWARD = 300; + CARAFE_FORWARD = 301; + COPY = 302; + DEFORM_ROI_POOL_BACKWARD = 400; + DEFORM_ROI_POOL_FORWARD = 401; + DIFF_IOU_ROTATED_SORT_VERTICES_FORWARD = 402; + DIV = 403; + DYNAMIC_POINT_TO_VOXEL_BACKWARD = 404; + DYNAMIC_POINT_TO_VOXEL_FORWARD = 405; + EXPAND = 500; + FILL = 600; + FOCAL_LOSS_SIGMOID_BACKWARD = 601; + FOCAL_LOSS_SIGMOID_FORWARD = 602; + GATHER_ND = 700; + GENERATE_PROPOSALS_V2 = 701; + GET_INDICE_PAIRS = 702; + INDICE_CONVOLUTION_BACKWARD_DATA = 900; + INDICE_CONVOLUTION_BACKWARD_FILTER = 901; + INDICE_CONVOLUTION_FORWARD = 902; + LOG = 1200; + MASKED_COL2IM_FORWARD = 1300; + MASKED_IM2COL_FORWARD = 1301; + MATMUL = 1302; + MOE_DISPATCH_BACKWARD_DATA = 1303; + MOE_DISPATCH_BACKWARD_GATE = 1304; + MOE_DISPATCH_FORWARD = 1305; + MS_DEFORM_ATTN_BACKWARD = 1306; + MS_DEFORM_ATTN_FORWARD = 1307; + MUTUAL_INFORMATION_BACKWARD = 1308; + MUTUAL_INFORMATION_FORWARD = 1309; + NMS = 1400; + NMS_ROTATED = 1401; + POINTS_IN_BOXES = 1600; + POLY_NMS = 1601; + PRIOR_BOX = 1602; + PSAMASK_BACKWARD = 1603; + PSAMASK_FORWARD = 1604; + PSROIPOOL_BACKWARD = 1605; + PSROIPOOL_FORWARD = 1606; + REDUCE = 1800; + ROI_ALIGN_BACKWARD = 1801; + ROI_ALIGN_ROTATED_BACKWARD = 1802; + ROI_ALIGN_ROTATED_FORWARD = 1803; + ROI_CROP_BACKWARD = 1804; + ROI_CROP_FORWARD = 1805; + ROI_POOLING_BACKWARD = 1806; + ROI_POOLING_FORWARD = 1807; + ROIALIGN_FORWARD = 1808; + ROIAWARE_POOL3D_BACKWARD = 1809; + ROIAWARE_POOL3D_FORWARD = 1810; + ROIPOINT_POOL3D = 1811; + ROTATED_FEATURE_ALIGN_BACKWARD = 1812; + ROTATED_FEATURE_ALIGN_FORWARD = 1813; + SCATTER_ND = 1900; + SQRT = 1901; + SQRT_BACKWARD = 1902; + STRIDED_SLICE = 1903; + SYNC_BATCH_NORM_BACKWARD_ELEMT = 1904; + SYNC_BATCHNORM_BACKWARD_ELEMT_V2 = 1905; + SYNC_BATCHNORM_BACKWARD_REDUCE = 1906; + SYNC_BATCHNORM_ELEMT = 1907; + SYNC_BATCHNORM_GATHER_STATS_WITH_COUNTS = 1908; + SYNC_BATCHNORM_STATS = 1909; + TENSOR_STRIDE_PROCESS = 2000; + THREE_INTERPOLATE_BACKWARD = 2001; + THREE_INTERPOLATE_FORWARD = 2002; + THREE_NN_FORWARD = 2003; + TIN_SHIFT_BACKWARD = 2004; + TIN_SHIFT_FORWARD = 2005; + TRANSFORM = 2006; + TRANSPOSE = 2007; + UNIQUE = 2100; + UTILS = 2101; + VOXEL_POOLING_FORWARD = 2200; + VOXELIZATION = 2201; + YOLO_BOX = 2500; +} + message Node { optional string valid = 1 [default = "True"]; optional string pattern = 2 [default = "random"]; @@ -237,7 +324,7 @@ message Node { required string op_name = 5 [default = "unknown"]; // op_name repeated Tensor input = 6; repeated Tensor output = 7; - optional string op_type = 8; // add op_type for mmcv op + optional OpType op_type = 8; // add op_type for mmcv op // evaluation optional TestParam test_param = 10; // remove this param later = delete