From a63a5735a3fee517d5ab7dcf677528bb90fe9544 Mon Sep 17 00:00:00 2001 From: Chuancysun <379087671@qq.com> Date: Sat, 16 Nov 2024 20:35:08 +0800 Subject: [PATCH 1/2] add LU parameter --- mlu_op_test.proto | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/mlu_op_test.proto b/mlu_op_test.proto index 3117d33..f4c2405 100755 --- a/mlu_op_test.proto +++ b/mlu_op_test.proto @@ -256,7 +256,8 @@ enum OpType { DYNAMIC_POINT_TO_VOXEL_BACKWARD = 404; DYNAMIC_POINT_TO_VOXEL_FORWARD = 405; EXPAND = 5; - FFT = 65536; + FFT = 65536; + SGETRF2 = 65537; FILL = 13199; FOCAL_LOSS_SIGMOID_BACKWARD = 12322; FOCAL_LOSS_SIGMOID_FORWARD = 15000; @@ -426,6 +427,7 @@ message Node { optional OpTensorParam op_tensor_param = 101; // OpTensorParam optional BatchMatMulBCastParam batch_matmul_bcast_param = 132; // param optional FFTParam fft_param = 65536; // param + optional Sgetrf2Param sgetrf2_param = 65537 ;//param optional DCNParam dcn_param = 405; optional LogcumsumexpParam logcumsumexp_param = 19980423; optional LgammaParam lgamma_param = 202435; @@ -1119,6 +1121,11 @@ message FFTParam { optional float scale_factor = 4 [default = 1.0]; } +// param to call Sgetrf2() +message Sgetrf2Param { + optional int32 mode =1 [default = 1]; +} + // param to call mluOpRoiPoolingForward() message RoiPoolingForwardParam { optional float spatial_scale = 1 [default = 1]; From b76f200af2077651ce54654c48209bfc95ad3e63 Mon Sep 17 00:00:00 2001 From: Chuancysun <379087671@qq.com> Date: Tue, 3 Dec 2024 00:20:16 +0800 Subject: [PATCH 2/2] [Fix](mluOpXgetrf):rename param --- mlu_op_test.proto | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mlu_op_test.proto b/mlu_op_test.proto index f4c2405..66f1c3a 100755 --- a/mlu_op_test.proto +++ b/mlu_op_test.proto @@ -427,7 +427,7 @@ message Node { optional OpTensorParam op_tensor_param = 101; // OpTensorParam optional BatchMatMulBCastParam batch_matmul_bcast_param = 132; // param optional FFTParam fft_param = 65536; // param - optional Sgetrf2Param sgetrf2_param = 65537 ;//param + optional XgetrfParam xgetrf_param = 65537 ;//param optional DCNParam dcn_param = 405; optional LogcumsumexpParam logcumsumexp_param = 19980423; optional LgammaParam lgamma_param = 202435; @@ -1121,8 +1121,8 @@ message FFTParam { optional float scale_factor = 4 [default = 1.0]; } -// param to call Sgetrf2() -message Sgetrf2Param { +// param to call Xgetrf() +message XgetrfParam { optional int32 mode =1 [default = 1]; }