From ec2378cdc96e79605f4f96fe744a4085ce693ada Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E6=A2=93=E4=B9=9D?= Date: Mon, 15 Jul 2024 16:37:29 +0800 Subject: [PATCH] fix tiling bug for mm/bmm example --- .../batch_matmul/kernel_launch_method_by_direct/main.cpp | 3 ++- examples/matrix/matmul/kernel_launch_method_by_direct/main.cpp | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/matrix/batch_matmul/kernel_launch_method_by_direct/main.cpp b/examples/matrix/batch_matmul/kernel_launch_method_by_direct/main.cpp index 504af2d2..5c057d03 100644 --- a/examples/matrix/batch_matmul/kernel_launch_method_by_direct/main.cpp +++ b/examples/matrix/batch_matmul/kernel_launch_method_by_direct/main.cpp @@ -9,6 +9,7 @@ */ #include "../../../common/data_utils.h" +#include "kernel_tiling/kernel_tiling.h" #ifndef ASCENDC_CPU_DEBUG #include "acl/acl.h" extern void BatchMatmulCustomDo(uint32_t coreDim, void* stream, @@ -63,7 +64,7 @@ int32_t main(int32_t argc, char *argv[]) size_t bFileSize = 64 * 1536 * sizeof(uint16_t); // uint16_t represent half size_t cFileSize = 192 * 256 * sizeof(float); uint32_t workspaceSize = 16 * 1024 * 1024; - size_t tilingFileSize = 48 * sizeof(uint32_t); + size_t tilingFileSize = sizeof(TCubeTiling); uint32_t blockDim = 1; #ifdef ASCENDC_CPU_DEBUG diff --git a/examples/matrix/matmul/kernel_launch_method_by_direct/main.cpp b/examples/matrix/matmul/kernel_launch_method_by_direct/main.cpp index ee8b9eb0..f245ac84 100644 --- a/examples/matrix/matmul/kernel_launch_method_by_direct/main.cpp +++ b/examples/matrix/matmul/kernel_launch_method_by_direct/main.cpp @@ -9,6 +9,7 @@ */ #include "../../../common/data_utils.h" +#include "kernel_tiling/kernel_tiling.h" #ifndef ASCENDC_CPU_DEBUG #include "acl/acl.h" @@ -64,7 +65,7 @@ int32_t main(int32_t argc, char *argv[]) size_t bFileSize = 512 * 1024 * sizeof(uint16_t); // uint16_t represent half size_t cFileSize = 512 * 1024 * sizeof(float); uint32_t workspaceSize = 16 * 1024 * 1024; - size_t tilingFileSize = 48 * sizeof(uint32_t); + size_t tilingFileSize = sizeof(TCubeTiling); #ifdef CUSTOM_ASCEND310P uint32_t blockDim = 2; #else -- Gitee