[Feature](mluOpLogcumsumexp): new parameters

Cambricon · Jun 24, 2024 · f3a0c55 · f3a0c55
1 parent dcef993
commit f3a0c55
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mlu_op_test.proto b/mlu_op_test.proto
@@ -426,6 +426,7 @@ message Node {
   optional BatchMatMulBCastParam batch_matmul_bcast_param       = 132;   // param
   optional FFTParam fft_param = 65536; // param
   optional DCNParam dcn_param       = 405;
+  optional LogcumsumexpParam logcumsumexp_param       = 998;
 }
 
 
@@ -1168,3 +1169,8 @@ message DCNParam {
   optional int32 im2col_step = 7;
   optional DataType compute_type = 8;
 }
+
+// param to call mluOpLogcumsumexp()
+message LogcumsumexpParam {
+  optional int32 dim = 1 [default = 0];
+}