From d10f9923a5661c78144a7207e7bfd469ceb5433d Mon Sep 17 00:00:00 2001 From: li-yuanjie-da Date: Thu, 25 Sep 2025 20:34:38 +0800 Subject: [PATCH] fix l0c_extend --- examples/OWNERS | 2 ++ .../op_kernel/matmul_l0c_extend_custom_kernel.h | 4 +++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/examples/OWNERS b/examples/OWNERS index 192a38db..8afc76b1 100644 --- a/examples/OWNERS +++ b/examples/OWNERS @@ -4,6 +4,8 @@ approvers: - sjtuhjh - mlidongfeng - ai-xin +- kong0808 +- suqwe reviewers: - wuzhaolinhuawei - horming diff --git a/examples/matrix/matmul_l0c_extend/op_kernel/matmul_l0c_extend_custom_kernel.h b/examples/matrix/matmul_l0c_extend/op_kernel/matmul_l0c_extend_custom_kernel.h index a2f68a71..53c962a2 100644 --- a/examples/matrix/matmul_l0c_extend/op_kernel/matmul_l0c_extend_custom_kernel.h +++ b/examples/matrix/matmul_l0c_extend/op_kernel/matmul_l0c_extend_custom_kernel.h @@ -41,12 +41,14 @@ public: */ __aicore__ inline void Process(); + static constexpr auto MM_CFG = GetNormalConfig(false, false, false, BatchMode::BATCH_LESS_THAN_L1, + true, IterateOrder::UNDEF, ScheduleType::INNER_PRODUCT, false); using A_TYPE = AscendC::MatmulType; using B_TYPE = AscendC::MatmulType; // C_TYPE TPosition is CO1, CubeFormat is NZ using C_TYPE = AscendC::MatmulType; using BIAS_TYPE = AscendC::MatmulType; - AscendC::Matmul matmulObj; + AscendC::Matmul matmulObj; private: /** -- Gitee