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 504af2d248ebddb07aea3a17f2a1a46c24e93c32..5c057d0320098480c78db61306dadc3e76b77023 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 ee8b9eb0b31b3b8224dbb15c1d61e39087caed40..f245ac84c9d39d593bd7b2b8f2db6089fb62bb9c 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