Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 5 commits into from
Oct 25, 2023
Merged

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

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
f75bc16
[Fix](bangc-ops): the type of op_type use enum instead of string
chqy99 Oct 18, 2023
868b3ab
[Fix](bangc-ops): the type of op_type use enum instead of string
chqy99 Oct 25, 2023
23d6aa1
[Fix](bangc-ops): the type of op_type use enum instead of string
chqy99 Oct 25, 2023
789569e
[Fix](bangc-ops): the type of op_type use enum instead of string
chqy99 Oct 25, 2023
d3e420f
[Fix](bangc-ops): the type of op_type use enum instead of string
chqy99 Oct 25, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
90 changes: 89 additions & 1 deletion mlu_op_test.proto
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -229,6 +229,94 @@ message SeqData {
optional RandomData random_data = 14;
}

enum OpType {
ABS = 2022;
ACTIVE_ROTATED_FILTER_FORWARD = 257073;
ADDN = 7178;
BALL_QUERY = 20231025;
BBOX_OVERLAPS = 2041;
BORDER_ALIGN_BACKWARD = 9001;
BORDER_ALIGN_FORWARD = 1009;
BOX_IOU_ROTATED = 200;
CARAFE_BACKWARD = 409;
CARAFE_FORWARD = 408;
COPY = 13;
DEFORM_ROI_POOL_BACKWARD = 10127;
DEFORM_ROI_POOL_FORWARD = 10126;
DIFF_IOU_ROTATED_SORT_VERTICES_FORWARD = 402;
DIV = 2020;
DYNAMIC_POINT_TO_VOXEL_BACKWARD = 404;
DYNAMIC_POINT_TO_VOXEL_FORWARD = 405;
EXPAND = 5;
FILL = 13199;
FOCAL_LOSS_SIGMOID_BACKWARD = 12322;
FOCAL_LOSS_SIGMOID_FORWARD = 15000;
GATHER_ND = 20;
GENERATE_PROPOSALS_V2 = 701;
GET_INDICE_PAIRS = 702;
INDICE_CONVOLUTION_BACKWARD_DATA = 900;
INDICE_CONVOLUTION_BACKWARD_FILTER = 901;
INDICE_CONVOLUTION_FORWARD = 902;
LOG = 20333;
MASKED_COL2IM_FORWARD = 2201;
MASKED_IM2COL_FORWARD = 2202;
MATMUL = 536470911;
MOE_DISPATCH_BACKWARD_DATA = 1303;
MOE_DISPATCH_BACKWARD_GATE = 1304;
MOE_DISPATCH_FORWARD = 1305;
MS_DEFORM_ATTN_BACKWARD = 1013;
MS_DEFORM_ATTN_FORWARD = 16005;
MUTUAL_INFORMATION_BACKWARD = 1308;
MUTUAL_INFORMATION_FORWARD = 1309;
NMS = 768;
NMS_ROTATED = 1401;
PAD = 33;
POINTS_IN_BOXES = 217;
POLY_NMS = 1602;
PRIOR_BOX = 1603;
PSAMASK_BACKWARD = 134659;
PSAMASK_FORWARD = 134658;
PSROIPOOL_BACKWARD = 1606;
PSROIPOOL_FORWARD = 1607;
REDUCE = 1199;
ROI_ALIGN_BACKWARD = 4427;
ROI_ALIGN_ROTATED_BACKWARD = 134661;
ROI_ALIGN_ROTATED_FORWARD = 134660;
ROI_CROP_BACKWARD = 1804;
ROI_CROP_FORWARD = 1805;
ROI_POOLING_BACKWARD = 888888;
ROI_POOLING_FORWARD = 210812;
ROIALIGN_FORWARD = 1808;
ROIAWARE_POOL3D_BACKWARD = 10121;
ROIAWARE_POOL3D_FORWARD = 10120;
ROIPOINT_POOL3D = 16002;
ROTATED_FEATURE_ALIGN_BACKWARD = 20221008;
ROTATED_FEATURE_ALIGN_FORWARD = 20229510;
SCATTER_ND = 359;
SQRT = 2021;
SQRT_BACKWARD = 55660;
STRIDED_SLICE = 657;
SYNC_BATCH_NORM_BACKWARD_ELEMT = 1663;
SYNC_BATCHNORM_BACKWARD_ELEMT_V2 = 1664;
SYNC_BATCHNORM_BACKWARD_REDUCE = 38324;
SYNC_BATCHNORM_ELEMT = 1662;
SYNC_BATCHNORM_GATHER_STATS_WITH_COUNTS = 11113;
SYNC_BATCHNORM_STATS = 11112;
TENSOR_STRIDE_PROCESS = 2000;
THREE_INTERPOLATE_BACKWARD = 2001;
THREE_INTERPOLATE_FORWARD = 2002;
THREE_NN_FORWARD = 2003;
TIN_SHIFT_BACKWARD = 150003;
TIN_SHIFT_FORWARD = 150002;
TRANSFORM = 4;
TRANSPOSE = 64;
UNIQUE = 1267;
UTILS = 2101;
VOXEL_POOLING_FORWARD = 2200;
VOXELIZATION = 2203;
YOLO_BOX = 2500;
}

message Node {
optional string valid = 1 [default = "True"];
optional string pattern = 2 [default = "random"];
Expand All @@ -237,7 +325,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;

// evaluation
optional TestParam test_param = 10; // remove this param later = delete
Expand Down