From c39402818a62a68ef935c3993ae1cfd8c8c956ab Mon Sep 17 00:00:00 2001 From: Nice_try <1849404617@qq.com> Date: Tue, 4 Jun 2024 02:55:50 +0000 Subject: [PATCH] =?UTF-8?q?fix=20comment=20on=20lp=5Fnorm=5Fv2=5Fcustom.cp?= =?UTF-8?q?p=20=E6=9B=B4=E6=94=B9=E6=B3=A8=E9=87=8A=EF=BC=8C=E5=A2=9E?= =?UTF-8?q?=E5=8A=A0=E7=BB=B4=E5=BA=A6=E4=B8=8D=E5=90=8C=E9=83=A8=E5=88=86?= =?UTF-8?q?=E4=BB=A3=E8=A1=A8=E7=9A=84=E5=85=B7=E4=BD=93=E5=90=AB=E4=B9=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Nice_try <1849404617@qq.com> --- .../LpNormV2Custom/op_host/lp_norm_v2_custom.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp b/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp index e7fd7aba0..b46a71d57 100644 --- a/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp +++ b/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp @@ -66,7 +66,9 @@ static ge::graphStatus TilingFunc(gert::TilingContext* context) // 处理多轴情况,axes为空时,不需要计算 // 仅处理了axes列表中为连续维度的情况,例如[0], [1, 2]之类 // 因为是连续维度,所以定位通过axes列表的值,将shape中的维度分为三部分 - // 前部分(unitCount),中间部分(stepSize),后部分(totalLength) + // unitCount 需要分别处理多少组元素 + // totalLength 每组要处理的元素的总个数 + // stepSize 每组中要处理的元素的相邻下标的距离 auto shape = context->GetInputTensor(0)->GetShape().GetOriginShape(); uint32_t dimNum = shape.GetDimNum(); uint32_t maxdim = 0; -- Gitee