add cholesky parameter

Cambricon · Apr 16, 2024 · 927342e · 927342e
1 parent 47aa527
commit 927342e
Showing 1 changed file with 7 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 CholeskyParam cholesky_param = 618323;  // CholeskyParam
 }
 
 
@@ -1177,3 +1178,9 @@ message DCNParam {
   optional int32 im2col_step = 7;
   optional DataType compute_type = 8;
 }
+
+//param to call mluOpCholesky()
+message CholeskyParam {
+  optional bool upper = 1 [default = false];
+}
+