[Fix](bangc-ops): the type of op_type use enum instead of string

Cambricon · Oct 18, 2023 · f75bc16 · f75bc16
1 parent 39b8c05
commit f75bc16
Showing 1 changed file with 88 additions and 1 deletion.
diff --git 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