diff --git a/mlu_op_test.proto b/mlu_op_test.proto
index 3117d33..62d52fb 100755
--- a/mlu_op_test.proto
+++ b/mlu_op_test.proto
@@ -328,6 +328,7 @@ enum OpType {
   DCN_BACKWARD_WEIGHT = 406;
   DCN_BACKWARD_DATA = 407;
   LGAMMA = 20243500;
+  CHOLESKY = 6183230;
 }
 
 message Node {
@@ -429,6 +430,7 @@ message Node {
   optional DCNParam dcn_param       = 405;
   optional LogcumsumexpParam logcumsumexp_param       = 19980423;
   optional LgammaParam lgamma_param = 202435;
+  optional CholeskyParam cholesky_param = 618323;  // CholeskyParam
 }
 
 
@@ -1191,3 +1193,9 @@ message LogcumsumexpParam {
 message LgammaParam {
   optional bool inplace = 1 [default = false];
 }
+
+//param to call mluOpCholesky()
+message CholeskyParam {
+  optional bool upper = 1 [default = false];
+} 
+