From b63ddfa6920f348f7a7a8820f66d68b8ed9a3f3a Mon Sep 17 00:00:00 2001 From: liuboxi Date: Wed, 13 Aug 2025 15:54:48 +0800 Subject: [PATCH] =?UTF-8?q?quant=E7=AE=97=E5=AD=90=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E6=9C=80=E5=90=8E=E4=B8=80=E7=BB=B432=E5=AD=97=E8=8A=82?= =?UTF-8?q?=E5=AF=B9=E9=BD=90=E7=9A=84=E6=A0=A1=E9=AA=8C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/kernels/kernels/elewise/quant/quant_tiling/quant_tiling.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/kernels/kernels/elewise/quant/quant_tiling/quant_tiling.cpp b/src/kernels/kernels/elewise/quant/quant_tiling/quant_tiling.cpp index 7ebefcb5..5431f956 100644 --- a/src/kernels/kernels/elewise/quant/quant_tiling/quant_tiling.cpp +++ b/src/kernels/kernels/elewise/quant/quant_tiling/quant_tiling.cpp @@ -33,6 +33,8 @@ Status QuantF16Tiling(const LaunchParam &launchParam, KernelInfo &kernelInfo) Status ret = PostLayerNormPtrFunc(tilingDataPtr, quantPtrCon, launchParam, kernelInfo); OP_TILING_CHECK_STATUS_RETURN(ret); int32_t scalarUsed = 256; + MKI_CHECK(quantPtrCon.numCol % 32 == 0, "last dim is not 32 bytes align", + return Status::FailStatus(ERROR_INVALID_VALUE, "last dim is not 32 bytes align")); // 32: 算子约束最后一维32字节对齐要求 MKI_CHECK(quantPtrCon.nlFirstdimPerCoreNum < (static_cast(std::numeric_limits::max()) / 2) / quantPtrCon.numCol, "numCol or nlFirstdimPerCoreNum is not invalid", -- Gitee