From b3685dd3ceb7dc00b8e8e72ac1bd6d11eb9a479f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=85=83=E6=9D=B0?= Date: Mon, 13 Oct 2025 06:34:31 +0000 Subject: [PATCH] modify bmm tiling --- .../host_tiling/basic_block_matmul_custom_tiling.h | 3 +-- .../basic_block_matmul_custom_tiling.cpp | 2 +- .../kernel_launch_method_by_direct/run.sh | 2 +- .../op_host/basic_block_matmul_custom.cpp | 2 +- .../batch_matmul/kernel_launch_method_by_direct/run.sh | 2 +- .../op_host/batch_matmul_bias_no_batch_custom_tiling.cpp | 7 ++----- examples/matrix/batch_matmul_bias_no_batch/run.sh | 2 +- .../op_host/batch_matmul_tscm_custom_tiling.cpp | 7 ++----- examples/matrix/batch_matmul_tscm/run.sh | 2 +- .../matrix/matmul/kernel_launch_method_by_direct/run.sh | 2 +- examples/matrix/matmul_a2b2share/run.sh | 2 +- examples/matrix/matmul_async_iterate/run.sh | 2 +- examples/matrix/matmul_async_iterate_all/run.sh | 2 +- examples/matrix/matmul_callback/run.sh | 2 +- examples/matrix/matmul_channelsplit/run.sh | 2 +- examples/matrix/matmul_constant/run.sh | 2 +- examples/matrix/matmul_gemv/run.sh | 2 +- examples/matrix/matmul_ibshareAB/run.sh | 2 +- examples/matrix/matmul_ibshareB/run.sh | 2 +- examples/matrix/matmul_int4/run.sh | 2 +- examples/matrix/matmul_k_reorder_load/run.sh | 2 +- examples/matrix/matmul_l0c_extend/run.sh | 2 +- examples/matrix/matmul_l0cache/run.sh | 2 +- examples/matrix/matmul_l2cache/run.sh | 2 +- examples/matrix/matmul_mixdualmaster/run.sh | 2 +- examples/matrix/matmul_mndb/run.sh | 2 +- examples/matrix/matmul_nbuffer33/run.sh | 2 +- examples/matrix/matmul_nd_align/run.sh | 2 +- examples/matrix/matmul_nz/run.sh | 2 +- examples/matrix/matmul_partial_output/run.sh | 2 +- examples/matrix/matmul_perf/run.sh | 2 +- examples/matrix/matmul_preload/run.sh | 2 +- examples/matrix/matmul_quant/run.sh | 2 +- examples/matrix/matmul_sparse/run.sh | 2 +- examples/matrix/matmul_splitk/run.sh | 2 +- examples/matrix/matmul_triangle/run.sh | 2 +- examples/matrix/matmul_tscm/run.sh | 2 +- examples/matrix/matmul_unaligned/run.sh | 2 +- examples/matrix/matmul_unitflag/run.sh | 2 +- 39 files changed, 41 insertions(+), 48 deletions(-) diff --git a/examples/matrix/basic_block_matmul/host_tiling/basic_block_matmul_custom_tiling.h b/examples/matrix/basic_block_matmul/host_tiling/basic_block_matmul_custom_tiling.h index b6e49cab..579112ab 100644 --- a/examples/matrix/basic_block_matmul/host_tiling/basic_block_matmul_custom_tiling.h +++ b/examples/matrix/basic_block_matmul/host_tiling/basic_block_matmul_custom_tiling.h @@ -21,11 +21,10 @@ END_TILING_DATA_DEF; REGISTER_TILING_DATA_CLASS(BasicBlockMatmulCustom, MatmulCustomTilingData) } -bool ComputeTiling(optiling::TCubeTiling& tiling, matmul_tiling::MultiCoreMatmulTiling* cubeTiling, +bool ComputeTiling(optiling::TCubeTiling& tiling, matmul_tiling::BatchMatmulTiling* cubeTiling, int32_t M, int32_t N, int32_t K,int32_t baseM, int32_t baseN, int32_t baseK, int32_t blockDim, bool isBias) { - cubeTiling->SetDim(blockDim); cubeTiling->SetAType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16, true); // A is transposed cubeTiling->SetBType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16); cubeTiling->SetCType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT); diff --git a/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/basic_block_matmul_custom_tiling.cpp b/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/basic_block_matmul_custom_tiling.cpp index 7fe7f1d8..b2d32058 100644 --- a/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/basic_block_matmul_custom_tiling.cpp +++ b/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/basic_block_matmul_custom_tiling.cpp @@ -40,7 +40,7 @@ uint8_t *GenerateTiling(int32_t M, int32_t N, int32_t K, int32_t baseM, int32_t { optiling::TCubeTiling tilingData; auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(socVersion); - matmul_tiling::MultiCoreMatmulTiling cubeTiling(*ascendcPlatform); + matmul_tiling::BatchMatmulTiling cubeTiling(*ascendcPlatform); bool res = ComputeTiling(tilingData, &cubeTiling, M, N, K, baseM, baseN, baseK, USED_CORE_NUM, false); if (!res) { diff --git a/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/run.sh b/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/run.sh index e4396318..818d52b4 100644 --- a/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/run.sh +++ b/examples/matrix/basic_block_matmul/kernel_launch_method_by_direct/run.sh @@ -42,7 +42,7 @@ if [ "${RUN_MODE}" = "sim" ]; then export LD_LIBRARY_PATH=$ASCEND_HOME_DIR/runtime/lib64/stub:$LD_LIBRARY_PATH fi -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/basic_block_matmul/kernel_launch_method_by_framework/op_host/basic_block_matmul_custom.cpp b/examples/matrix/basic_block_matmul/kernel_launch_method_by_framework/op_host/basic_block_matmul_custom.cpp index e5ac288e..813cab42 100644 --- a/examples/matrix/basic_block_matmul/kernel_launch_method_by_framework/op_host/basic_block_matmul_custom.cpp +++ b/examples/matrix/basic_block_matmul/kernel_launch_method_by_framework/op_host/basic_block_matmul_custom.cpp @@ -26,7 +26,7 @@ static ge::graphStatus TilingFunc(gert::TilingContext* context) int32_t baseN = 256; int32_t baseK = 64; bool isBias = true; - matmul_tiling::MultiCoreMatmulTiling cubeTiling(ascendcPlatform); + matmul_tiling::BatchMatmulTiling cubeTiling(ascendcPlatform); int blockDim = USED_CORE_NUM * 2; // AIV SetDim MatmulCustomTilingData tiling; bool res = ComputeTiling(tiling.cubeTilingData, &cubeTiling, M, N, K, baseM, baseN, baseK, blockDim, isBias); diff --git a/examples/matrix/batch_matmul/kernel_launch_method_by_direct/run.sh b/examples/matrix/batch_matmul/kernel_launch_method_by_direct/run.sh index a95e5d1b..7c7b0d07 100644 --- a/examples/matrix/batch_matmul/kernel_launch_method_by_direct/run.sh +++ b/examples/matrix/batch_matmul/kernel_launch_method_by_direct/run.sh @@ -50,7 +50,7 @@ if [ "${RUN_MODE}" = "sim" ]; then export LD_LIBRARY_PATH=$ASCEND_HOME_DIR/runtime/lib64/stub:$LD_LIBRARY_PATH fi -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/batch_matmul_bias_no_batch/op_host/batch_matmul_bias_no_batch_custom_tiling.cpp b/examples/matrix/batch_matmul_bias_no_batch/op_host/batch_matmul_bias_no_batch_custom_tiling.cpp index bb80ce11..69a5857f 100644 --- a/examples/matrix/batch_matmul_bias_no_batch/op_host/batch_matmul_bias_no_batch_custom_tiling.cpp +++ b/examples/matrix/batch_matmul_bias_no_batch/op_host/batch_matmul_bias_no_batch_custom_tiling.cpp @@ -17,7 +17,7 @@ TCubeTiling GenerateTiling(const MatmulCaseParams& testCaseParams) { TCubeTiling tilingData; auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); - matmul_tiling::MultiCoreMatmulTiling cubeTiling(*ascendcPlatform); + matmul_tiling::BatchMatmulTiling cubeTiling(*ascendcPlatform); int32_t M = testCaseParams.m; int32_t N = testCaseParams.n; int32_t K = testCaseParams.k; @@ -27,7 +27,6 @@ TCubeTiling GenerateTiling(const MatmulCaseParams& testCaseParams) bool isBtrans = testCaseParams.isBTrans; int32_t batchNum = testCaseParams.batchNum; - cubeTiling.SetDim(blockDim); cubeTiling.SetAType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16, isAtrans); cubeTiling.SetBType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, @@ -40,9 +39,7 @@ TCubeTiling GenerateTiling(const MatmulCaseParams& testCaseParams) cubeTiling.SetShape(M, N, K); cubeTiling.EnableBias(isBias); cubeTiling.SetBufferSpace(-1, -1, -1); - cubeTiling.SetALayout(batchNum, M, 1, 1, K); - cubeTiling.SetBLayout(batchNum, K, 1, 1, N); - cubeTiling.SetCLayout(batchNum, M, 1, 1, N); + cubeTiling.SetBatchInfoForNormal(batchNum, batchNum, M, N, K); cubeTiling.SetBatchNum(batchNum); if (cubeTiling.GetTiling(tilingData) == -1) { std::cout << "Generate tiling failed." << std::endl; diff --git a/examples/matrix/batch_matmul_bias_no_batch/run.sh b/examples/matrix/batch_matmul_bias_no_batch/run.sh index 4652cec2..dd35b30a 100644 --- a/examples/matrix/batch_matmul_bias_no_batch/run.sh +++ b/examples/matrix/batch_matmul_bias_no_batch/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/batch_matmul_tscm/op_host/batch_matmul_tscm_custom_tiling.cpp b/examples/matrix/batch_matmul_tscm/op_host/batch_matmul_tscm_custom_tiling.cpp index 3b167ecd..5bfbd0f2 100644 --- a/examples/matrix/batch_matmul_tscm/op_host/batch_matmul_tscm_custom_tiling.cpp +++ b/examples/matrix/batch_matmul_tscm/op_host/batch_matmul_tscm_custom_tiling.cpp @@ -17,7 +17,7 @@ TCubeTiling GenerateTiling(const MatmulCaseParams& testCaseParams) { TCubeTiling tilingData; auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); - matmul_tiling::MultiCoreMatmulTiling cubeTiling(*ascendcPlatform); + matmul_tiling::BatchMatmulTiling cubeTiling(*ascendcPlatform); int32_t M = testCaseParams.m; int32_t N = testCaseParams.n; int32_t K = testCaseParams.k; @@ -27,7 +27,6 @@ TCubeTiling GenerateTiling(const MatmulCaseParams& testCaseParams) bool isBtrans = testCaseParams.isBTrans; int32_t batchNum = testCaseParams.batchNum; - cubeTiling.SetDim(blockDim); cubeTiling.SetAType(matmul_tiling::TPosition::TSCM, matmul_tiling::CubeFormat::NZ, matmul_tiling::DataType::DT_FLOAT16, isAtrans); cubeTiling.SetBType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, @@ -40,9 +39,7 @@ TCubeTiling GenerateTiling(const MatmulCaseParams& testCaseParams) cubeTiling.SetShape(M, N, K); cubeTiling.EnableBias(isBias); cubeTiling.SetBufferSpace(-1, -1, -1); - cubeTiling.SetALayout(batchNum, M, 1, 1, K); - cubeTiling.SetBLayout(batchNum, K, 1, 1, N); - cubeTiling.SetCLayout(batchNum, M, 1, 1, N); + cubeTiling.SetBatchInfoForNormal(batchNum, batchNum, M, N, K); cubeTiling.SetBatchNum(batchNum); if (cubeTiling.GetTiling(tilingData) == -1) { std::cout << "Generate tiling failed." << std::endl; diff --git a/examples/matrix/batch_matmul_tscm/run.sh b/examples/matrix/batch_matmul_tscm/run.sh index aec9d4ce..e013418a 100644 --- a/examples/matrix/batch_matmul_tscm/run.sh +++ b/examples/matrix/batch_matmul_tscm/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul/kernel_launch_method_by_direct/run.sh b/examples/matrix/matmul/kernel_launch_method_by_direct/run.sh index d7c77e32..102d37f0 100644 --- a/examples/matrix/matmul/kernel_launch_method_by_direct/run.sh +++ b/examples/matrix/matmul/kernel_launch_method_by_direct/run.sh @@ -54,7 +54,7 @@ if [ ! "${CASE_TYPE_DIRECT}" ]; then CASE_TYPE_DIRECT=align fi -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_a2b2share/run.sh b/examples/matrix/matmul_a2b2share/run.sh index 37b063b4..9672881e 100644 --- a/examples/matrix/matmul_a2b2share/run.sh +++ b/examples/matrix/matmul_a2b2share/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH export ASCEND_TOOLKIT_HOME=${ASCEND_HOME_DIR} set -euo pipefail diff --git a/examples/matrix/matmul_async_iterate/run.sh b/examples/matrix/matmul_async_iterate/run.sh index 11ff84f6..96297b10 100644 --- a/examples/matrix/matmul_async_iterate/run.sh +++ b/examples/matrix/matmul_async_iterate/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH export ASCEND_TOOLKIT_HOME=${ASCEND_HOME_DIR} set -euo pipefail diff --git a/examples/matrix/matmul_async_iterate_all/run.sh b/examples/matrix/matmul_async_iterate_all/run.sh index e42cc839..31f8da34 100644 --- a/examples/matrix/matmul_async_iterate_all/run.sh +++ b/examples/matrix/matmul_async_iterate_all/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH export ASCEND_TOOLKIT_HOME=${ASCEND_HOME_DIR} set -euo pipefail diff --git a/examples/matrix/matmul_callback/run.sh b/examples/matrix/matmul_callback/run.sh index 1e716556..e913e2ee 100644 --- a/examples/matrix/matmul_callback/run.sh +++ b/examples/matrix/matmul_callback/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_channelsplit/run.sh b/examples/matrix/matmul_channelsplit/run.sh index 29eb47af..d9119791 100644 --- a/examples/matrix/matmul_channelsplit/run.sh +++ b/examples/matrix/matmul_channelsplit/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_constant/run.sh b/examples/matrix/matmul_constant/run.sh index 5c05c655..2d842ab5 100644 --- a/examples/matrix/matmul_constant/run.sh +++ b/examples/matrix/matmul_constant/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_gemv/run.sh b/examples/matrix/matmul_gemv/run.sh index 75d2dd48..03443236 100644 --- a/examples/matrix/matmul_gemv/run.sh +++ b/examples/matrix/matmul_gemv/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_ibshareAB/run.sh b/examples/matrix/matmul_ibshareAB/run.sh index 746f8655..e4c0478f 100644 --- a/examples/matrix/matmul_ibshareAB/run.sh +++ b/examples/matrix/matmul_ibshareAB/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_ibshareB/run.sh b/examples/matrix/matmul_ibshareB/run.sh index f38c0f87..7b7cd9bd 100644 --- a/examples/matrix/matmul_ibshareB/run.sh +++ b/examples/matrix/matmul_ibshareB/run.sh @@ -77,7 +77,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_int4/run.sh b/examples/matrix/matmul_int4/run.sh index fb73e320..b4dc6911 100644 --- a/examples/matrix/matmul_int4/run.sh +++ b/examples/matrix/matmul_int4/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_k_reorder_load/run.sh b/examples/matrix/matmul_k_reorder_load/run.sh index a53e37c1..93fe4860 100644 --- a/examples/matrix/matmul_k_reorder_load/run.sh +++ b/examples/matrix/matmul_k_reorder_load/run.sh @@ -75,7 +75,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_l0c_extend/run.sh b/examples/matrix/matmul_l0c_extend/run.sh index 85a36eb1..7c45ea25 100644 --- a/examples/matrix/matmul_l0c_extend/run.sh +++ b/examples/matrix/matmul_l0c_extend/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_l0cache/run.sh b/examples/matrix/matmul_l0cache/run.sh index 331d8a0b..d792b5c7 100644 --- a/examples/matrix/matmul_l0cache/run.sh +++ b/examples/matrix/matmul_l0cache/run.sh @@ -50,7 +50,7 @@ if [ "${RUN_MODE}" = "sim" ]; then export LD_LIBRARY_PATH=$ASCEND_HOME_DIR/runtime/lib64/stub:$LD_LIBRARY_PATH fi -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_l2cache/run.sh b/examples/matrix/matmul_l2cache/run.sh index 5c05c655..2d842ab5 100644 --- a/examples/matrix/matmul_l2cache/run.sh +++ b/examples/matrix/matmul_l2cache/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_mixdualmaster/run.sh b/examples/matrix/matmul_mixdualmaster/run.sh index 53d52d17..e5bf4723 100644 --- a/examples/matrix/matmul_mixdualmaster/run.sh +++ b/examples/matrix/matmul_mixdualmaster/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_mndb/run.sh b/examples/matrix/matmul_mndb/run.sh index c378e7c6..52a6c56c 100644 --- a/examples/matrix/matmul_mndb/run.sh +++ b/examples/matrix/matmul_mndb/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_nbuffer33/run.sh b/examples/matrix/matmul_nbuffer33/run.sh index 5b5887f5..0f447516 100644 --- a/examples/matrix/matmul_nbuffer33/run.sh +++ b/examples/matrix/matmul_nbuffer33/run.sh @@ -75,7 +75,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_nd_align/run.sh b/examples/matrix/matmul_nd_align/run.sh index 9f9ec6fa..a80f0d73 100644 --- a/examples/matrix/matmul_nd_align/run.sh +++ b/examples/matrix/matmul_nd_align/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_nz/run.sh b/examples/matrix/matmul_nz/run.sh index 5c05c655..2d842ab5 100644 --- a/examples/matrix/matmul_nz/run.sh +++ b/examples/matrix/matmul_nz/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_partial_output/run.sh b/examples/matrix/matmul_partial_output/run.sh index 68be88e5..33961a2a 100644 --- a/examples/matrix/matmul_partial_output/run.sh +++ b/examples/matrix/matmul_partial_output/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_perf/run.sh b/examples/matrix/matmul_perf/run.sh index c090d308..34be7e88 100644 --- a/examples/matrix/matmul_perf/run.sh +++ b/examples/matrix/matmul_perf/run.sh @@ -80,7 +80,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_preload/run.sh b/examples/matrix/matmul_preload/run.sh index 66d3a53f..d093d3d1 100644 --- a/examples/matrix/matmul_preload/run.sh +++ b/examples/matrix/matmul_preload/run.sh @@ -82,7 +82,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_quant/run.sh b/examples/matrix/matmul_quant/run.sh index 6c7e0950..29d7f6f1 100644 --- a/examples/matrix/matmul_quant/run.sh +++ b/examples/matrix/matmul_quant/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_sparse/run.sh b/examples/matrix/matmul_sparse/run.sh index 53a29cf4..da2b911b 100644 --- a/examples/matrix/matmul_sparse/run.sh +++ b/examples/matrix/matmul_sparse/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_splitk/run.sh b/examples/matrix/matmul_splitk/run.sh index 168c410b..b1bb8a1f 100644 --- a/examples/matrix/matmul_splitk/run.sh +++ b/examples/matrix/matmul_splitk/run.sh @@ -75,7 +75,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_triangle/run.sh b/examples/matrix/matmul_triangle/run.sh index e9c582da..4fe12425 100644 --- a/examples/matrix/matmul_triangle/run.sh +++ b/examples/matrix/matmul_triangle/run.sh @@ -74,7 +74,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH export ASCEND_TOOLKIT_HOME=${ASCEND_HOME_DIR} set -euo pipefail diff --git a/examples/matrix/matmul_tscm/run.sh b/examples/matrix/matmul_tscm/run.sh index b6f86c96..30fa4267 100644 --- a/examples/matrix/matmul_tscm/run.sh +++ b/examples/matrix/matmul_tscm/run.sh @@ -66,7 +66,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail diff --git a/examples/matrix/matmul_unaligned/run.sh b/examples/matrix/matmul_unaligned/run.sh index 3b607bf8..e8ea2454 100644 --- a/examples/matrix/matmul_unaligned/run.sh +++ b/examples/matrix/matmul_unaligned/run.sh @@ -64,7 +64,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH export ASCEND_TOOLKIT_HOME=${ASCEND_HOME_DIR} export TF_CPP_MIN_LOG_LEVEL=3 diff --git a/examples/matrix/matmul_unitflag/run.sh b/examples/matrix/matmul_unitflag/run.sh index 23cf0e1d..1b82eb5a 100644 --- a/examples/matrix/matmul_unitflag/run.sh +++ b/examples/matrix/matmul_unitflag/run.sh @@ -75,7 +75,7 @@ rm -rf build mkdir build cd build -source $ASCEND_HOME_DIR/bin/setenv.bash +export PATH=${ASCEND_HOME_DIR}/tools/msprof/:$PATH export LD_LIBRARY_PATH=${ASCEND_HOME_DIR}/tools/simulator/${SOC_VERSION}/lib:$LD_LIBRARY_PATH set -euo pipefail -- Gitee