From 84dfa8001df9841f79417a086b7bf3fc56848eb8 Mon Sep 17 00:00:00 2001
From: leg190 <75871074+leg190@users.noreply.github.com>
Date: Sun, 29 Sep 2024 15:17:44 +0800
Subject: [PATCH] [Feature](mluOpLogspace): new parameters (#103)

---
 mlu_op_test.proto | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/mlu_op_test.proto b/mlu_op_test.proto
index da7d42c..ba3b896 100755
--- a/mlu_op_test.proto
+++ b/mlu_op_test.proto
@@ -267,6 +267,7 @@ enum OpType {
   INDICE_CONVOLUTION_BACKWARD_FILTER = 901;
   INDICE_CONVOLUTION_FORWARD = 902;
   LOG = 20333;
+  LOGSPACE = 29299;
   MASKED_COL2IM_FORWARD = 2201;
   MASKED_IM2COL_FORWARD = 2202;
   MATMUL = 536470911;
@@ -365,6 +366,7 @@ message Node {
   optional DivParam    div_param                                = 4001;   // param
   optional ReduceParam reduce_param                             = 85;     // param
   optional LogParam    log_param                                = 4002;   // param
+  optional LogspaceParam logspace_param                         = 20240307; //param
   optional SqrtParam   sqrt_param                               = 4003;   // param
   optional PsRoiPoolForwardParam psroipool_forward_param        = 4004;   // param
   optional PsRoiPoolBackwardParam psroipool_backward_param      = 4005;   // param
@@ -643,6 +645,14 @@ message LogParam {
   optional ComputationPreference prefer = 2 [default = COMPUTATION_HIGH_PRECISION];
 }
 
+// param to call mluOpLogspace()
+message LogspaceParam {
+  required float start        = 1 [default = 1.0];
+  required float end        = 2 [default = 3.0];
+  required int64 steps        = 3 [default = 1024];
+  required float base        = 4 [default = 10.0];
+}
+
 // param to call mluOpNms()
 message NmsParam{
   required int32 run_mode = 1 [default = 0];