From 99ec0dcb2a72be1feced1ad4158be0914ecf693b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E6=A2=93=E4=B9=9D?= Date: Tue, 16 Jul 2024 14:32:09 +0800 Subject: [PATCH] fix tiling socversion bug --- .../matrix/matmul/kernel_launch_method_by_direct/main.cpp | 4 ++-- .../kernel_launch_method_by_direct/matmul_custom_tiling.cpp | 6 ++++-- 2 files changed, 6 insertions(+), 4 deletions(-) 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 f245ac84..ce8f6ba8 100644 --- a/examples/matrix/matmul/kernel_launch_method_by_direct/main.cpp +++ b/examples/matrix/matmul/kernel_launch_method_by_direct/main.cpp @@ -67,9 +67,9 @@ int32_t main(int32_t argc, char *argv[]) uint32_t workspaceSize = 16 * 1024 * 1024; size_t tilingFileSize = sizeof(TCubeTiling); #ifdef CUSTOM_ASCEND310P - uint32_t blockDim = 2; + uint32_t blockDim = 4; #else - uint32_t blockDim = 1; + uint32_t blockDim = 2; #endif #ifdef ASCENDC_CPU_DEBUG diff --git a/examples/matrix/matmul/kernel_launch_method_by_direct/matmul_custom_tiling.cpp b/examples/matrix/matmul/kernel_launch_method_by_direct/matmul_custom_tiling.cpp index ef3a826e..21387beb 100644 --- a/examples/matrix/matmul/kernel_launch_method_by_direct/matmul_custom_tiling.cpp +++ b/examples/matrix/matmul/kernel_launch_method_by_direct/matmul_custom_tiling.cpp @@ -14,11 +14,12 @@ #include #include #include "tiling/tiling_api.h" +#include "tiling/platform/platform_ascendc.h" #include "../host_tiling/matmul_custom_tiling.h" using namespace matmul_tiling; using namespace std; -constexpr int32_t USED_CORE_NUM = 2; +constexpr int32_t USED_CORE_NUM = 4; uint8_t *GetTilingBuf(optiling::TCubeTiling *tilingData) { @@ -35,7 +36,8 @@ uint8_t *GenerateTiling() int K = 512; bool isBias = false; optiling::TCubeTiling tilingData; - MultiCoreMatmulTiling cubeTiling; + auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); + MultiCoreMatmulTiling cubeTiling(*ascendcPlatform); bool res = ComputeTiling(tilingData, &cubeTiling, M, N, K, USED_CORE_NUM, false); if (!res) { cout << "gen tiling failed" << endl; -- Gitee