From 7e8a5d2e07e19fae3aa18771eec4d194a1fcaf34 Mon Sep 17 00:00:00 2001 From: chentianyu19 Date: Mon, 15 Dec 2025 10:47:02 +0800 Subject: [PATCH] Fix matmul --- .../0_introduction/26_simple_matmulleakyrelu/CMakeLists.txt | 1 + .../26_simple_matmulleakyrelu/matmul_leakyrelu.asc | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/CMakeLists.txt b/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/CMakeLists.txt index 05004ccca..44b107442 100644 --- a/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/CMakeLists.txt +++ b/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/CMakeLists.txt @@ -13,6 +13,7 @@ target_link_libraries(demo PRIVATE register platform m + dl ) target_compile_options(demo PRIVATE diff --git a/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/matmul_leakyrelu.asc b/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/matmul_leakyrelu.asc index 7cdcc247f..be3ea10ea 100644 --- a/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/matmul_leakyrelu.asc +++ b/operator/ascendc/0_introduction/26_simple_matmulleakyrelu/matmul_leakyrelu.asc @@ -166,11 +166,11 @@ MatmulLeakyKernel::CalcOffset(int32_t blockIdx, c * @param bias: Bias gm addr. * @param c: Out gm addr. * @param workspace: Temporary gm space addr required by matmul calc. - * @param tiling: Tiling data. + * @param tiling: Tiling data. * @retval None */ __global__ __aicore__ void matmul_leakyrelu_custom(GM_ADDR a, GM_ADDR b, GM_ADDR bias, GM_ADDR c, - GM_ADDR workspace, AscendC::tiling::TCubeTiling tiling) + __kfc_workspace__ GM_ADDR workspace, AscendC::tiling::TCubeTiling tiling) { KERNEL_TASK_TYPE_DEFAULT(KERNEL_TYPE_MIX_AIC_1_2); AscendC::TPipe pipe; @@ -182,7 +182,7 @@ __global__ __aicore__ void matmul_leakyrelu_custom(GM_ADDR a, GM_ADDR b, GM_ADDR /** * @brief Generate matmul tiling. - * @param ascendcPlatform: platform info. + * @param ascendcPlatform: platform info. */ AscendC::tiling::TCubeTiling GenerateTiling(platform_ascendc::PlatformAscendC* ascendcPlatform) { -- Gitee