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 f245ac84c9d39d593bd7b2b8f2db6089fb62bb9c..ce8f6ba8599673954410e0679d6e949af181dccd 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 ef3a826e351ff5fd42201f3558f65e8d0ded98f7..21387bebb30fc75194c548a167eb98b2b6d63715 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;