From 7d348601b0d5cd7713be5ef7dd8453495696c5bc Mon Sep 17 00:00:00 2001
From: yuantao <taoyuan15@h-partners.com>
Date: Wed, 16 Jul 2025 07:26:54 +0000
Subject: [PATCH 1/7] support simulator

---
 .../CMakeDetermineASCENDCompiler.cmake        | 24 ++++++++++++++++++-
 examples/CMakeLists.txt                       |  8 +++++--
 scripts/build.sh                              | 22 ++++++++++++++++-
 3 files changed, 50 insertions(+), 4 deletions(-)

diff --git a/cmake/modules/CMakeDetermineASCENDCompiler.cmake b/cmake/modules/CMakeDetermineASCENDCompiler.cmake
index 72b7563f..84c8a927 100644
--- a/cmake/modules/CMakeDetermineASCENDCompiler.cmake
+++ b/cmake/modules/CMakeDetermineASCENDCompiler.cmake
@@ -33,9 +33,31 @@ set(_CMAKE_ASCEND_HOST_IMPLICIT_LINK_DIRECTORIES
 
 set(_CMAKE_ASCEND_HOST_IMPLICIT_LINK_LIBRARIES
     stdc++
-    runtime
 )
 
+if(DEFINED ENABLE_SIMULATOR AND ENABLE_SIMULATOR)
+    if(NOT DEFINED SIMULATOR_NPU_MODEL)
+        message(WARNING "Simulator mode is enabled but SIMULATOR_NPU_MODEL is not defined. Try get model from LD_LIBRARY_PATH.")
+        set(LD_LIBRARY_PATH $ENV{LD_LIBRARY_PATH})
+        string(REGEX MATCH "simulator/([^:/]*)" SUBDIR "${LD_LIBRARY_PATH}")
+        if(SUBDIR)
+            set(SIMULATOR_NPU_MODEL "${CMAKE_MATCH_1}")
+            message(STATUS "Matched SIMULATOR_NPU_MODEL: ${SIMULATOR_NPU_MODEL}")
+        else()
+            message(FATAL_ERROR "No SIMULATOR_NPU_MODEL matched!")
+        endif()
+    endif()
+
+    list(APPEND _CMAKE_ASCEND_HOST_IMPLICIT_LINK_DIRECTORIES
+        ${_CMAKE_ASCEND_HOME_PATH}/tools/simulator/${SIMULATOR_NPU_MODEL}/lib
+        ${_CMAKE_ASCEND_HOME_PATH}/acllib/lib64/stub)
+    list(APPEND _CMAKE_ASCEND_HOST_IMPLICIT_LINK_LIBRARIES
+        runtime_camodel)
+else()
+    list(APPEND _CMAKE_ASCEND_HOST_IMPLICIT_LINK_LIBRARIES
+        runtime)
+endif()
+
 if(DEFINED ENABLE_MSPROF AND ENABLE_MSPROF)
     list(APPEND _CMAKE_ASCEND_HOST_IMPLICIT_LINK_LIBRARIES profapi)
 endif()
diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt
index cd138646..8743681b 100644
--- a/examples/CMakeLists.txt
+++ b/examples/CMakeLists.txt
@@ -11,11 +11,15 @@ set(CATLASS_EXAMPLES_COMMON_SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/common)
 add_custom_target(catlass_examples)
 
 macro(catlass_example_add_executable NAME AICORE_ARCH)
+    set(EXAMPLE_DESTINATION bin)
+    if(DEFINED ENABLE_SIMULATOR AND ENABLE_SIMULATOR)
+        set(EXAMPLE_DESTINATION bin_sim)
+    endif()
     add_executable(${NAME} ${ARGN})
     target_compile_options(${NAME} PRIVATE $<$<COMPILE_LANGUAGE:ASCEND>:--cce-aicore-arch=${AICORE_ARCH}>)
     add_dependencies(catlass_examples ${NAME})
-    install(TARGETS ${NAME} DESTINATION bin COMPONENT ${NAME})
-    install(TARGETS ${NAME} DESTINATION bin COMPONENT catlass_examples)
+    install(TARGETS ${NAME} DESTINATION ${EXAMPLE_DESTINATION} COMPONENT ${NAME})
+    install(TARGETS ${NAME} DESTINATION ${EXAMPLE_DESTINATION} COMPONENT catlass_examples)
 endmacro()
 
 add_compile_options(
diff --git a/scripts/build.sh b/scripts/build.sh
index b32152c2..adf27fb5 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -12,6 +12,16 @@ set -o errexit
 set -o nounset
 set -o pipefail
 
+function get_npu_model(){
+    if command -v npu-smi &> /dev/null; then
+        echo "Ascend$(npu-smi info -t board -i 0 -c 0 | grep \"Chip Name\" | awk '{print $NF}')"
+        return 1
+    else
+        echo "Ascend910B1"
+        return 0
+    fi
+}
+
 RED='\033[0;31m'
 GREEN='\033[0;32m'
 YELLOW='\033[0;33m'
@@ -48,6 +58,7 @@ TARGET=""
 CMAKE_BUILD_TYPE="Release"
 declare -a CMAKE_OPTIONS=()
 CLEAN=false
+POST_BUILD_HINT=""
 
 while [[ $# -gt 0 ]]; do
     case "$1" in
@@ -65,6 +76,13 @@ while [[ $# -gt 0 ]]; do
         --msdebug)
             CMAKE_OPTIONS+=("-DENABLE_MSDEBUG=True")
             ;;
+        --simulator)
+            CMAKE_OPTIONS+=("-DENABLE_SIMULATOR=True")
+            if NPU_MODEL=$(get_npu_model); then
+                echo -e "${YELLOW}No npu-smi detected, using default model for simulator: ${NPU_MODEL}${NC}"
+            fi
+            CMAKE_OPTIONS+=("-DSIMULATOR_NPU_MODEL=${NPU_MODEL}")
+            POST_BUILD_HINT="Please run \nexport LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/tools/simulator/${NPU_MODEL}/lib:\$LD_LIBRARY_PATH\nin your terminal before execute examples."            ;;
         --enable_msprof)
             CMAKE_OPTIONS+=("-DENABLE_MSPROF=True")
             ;;
@@ -137,4 +155,6 @@ case "$TARGET" in
         cmake --install "$BUILD_DIR" --component "$TARGET"
         echo -e "${GREEN}Target '$TARGET' built successfully${NC}"
         ;;
-esac
\ No newline at end of file
+esac
+
+echo -e "$POST_BUILD_HINT"
\ No newline at end of file
-- 
Gitee


From 7e9b469e3db25190997aaa5444ac782a54ebb6a7 Mon Sep 17 00:00:00 2001
From: yuantao <taoyuan15@h-partners.com>
Date: Wed, 16 Jul 2025 07:27:49 +0000
Subject: [PATCH 2/7] support simulator

---
 scripts/build.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scripts/build.sh b/scripts/build.sh
index adf27fb5..90655a91 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -45,6 +45,7 @@ function show_help() {
     echo "  --clean         Clean build directories"
     echo "  --debug         Build in debug mode"
     echo "  --msdebug       Enable msdebug support"
+    echo "  --simulator     Compile example in simulator mode"
     echo "  --enable_msprof Enable msprofiling"
     echo "  -D<option>      Additional CMake options"
     echo -e "\n${BLUE}Targets:${NC}"
-- 
Gitee


From 1e11298b4f656e53af6ded07440f3ff9a249943b Mon Sep 17 00:00:00 2001
From: yuantao <taoyuan15@h-partners.com>
Date: Wed, 16 Jul 2025 07:34:28 +0000
Subject: [PATCH 3/7] support simulator

---
 scripts/build.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/build.sh b/scripts/build.sh
index 90655a91..65fc1dd7 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -14,7 +14,7 @@ set -o pipefail
 
 function get_npu_model(){
     if command -v npu-smi &> /dev/null; then
-        echo "Ascend$(npu-smi info -t board -i 0 -c 0 | grep \"Chip Name\" | awk '{print $NF}')"
+        echo "Ascend$(npu-smi info -t board -i 0 -c 0 | awk '/Chip Name/ {print $NF}')"
         return 1
     else
         echo "Ascend910B1"
-- 
Gitee


From 48f5c16a9caf7323eb9c71a7f7f387a5755c1e7f Mon Sep 17 00:00:00 2001
From: yuantao <taoyuan15@h-partners.com>
Date: Wed, 16 Jul 2025 07:37:07 +0000
Subject: [PATCH 4/7] support simulator

---
 scripts/build.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/build.sh b/scripts/build.sh
index 65fc1dd7..3d54fb59 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -81,6 +81,8 @@ while [[ $# -gt 0 ]]; do
             CMAKE_OPTIONS+=("-DENABLE_SIMULATOR=True")
             if NPU_MODEL=$(get_npu_model); then
                 echo -e "${YELLOW}No npu-smi detected, using default model for simulator: ${NPU_MODEL}${NC}"
+            else
+                echo -e "${GREEN} Detect NPU_MODEL: ${NPU_MODEL}"
             fi
             CMAKE_OPTIONS+=("-DSIMULATOR_NPU_MODEL=${NPU_MODEL}")
             POST_BUILD_HINT="Please run \nexport LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/tools/simulator/${NPU_MODEL}/lib:\$LD_LIBRARY_PATH\nin your terminal before execute examples."            ;;
-- 
Gitee


From f8bcbbdea77ae9ce10c5edf599b1285bb961191e Mon Sep 17 00:00:00 2001
From: yuantao <taoyuan15@h-partners.com>
Date: Wed, 16 Jul 2025 08:34:11 +0000
Subject: [PATCH 5/7] update cann check

---
 scripts/build.sh | 66 ++++++++++++++++++++++++++++--------------------
 1 file changed, 39 insertions(+), 27 deletions(-)

diff --git a/scripts/build.sh b/scripts/build.sh
index 3d54fb59..e5aff6f2 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -12,6 +12,16 @@ set -o errexit
 set -o nounset
 set -o pipefail
 
+NC="\033[0m"
+RED="\033[0;31m"
+GREEN="\033[0;32m"
+YELLOW="\033[0;33m"
+BLUE="\033[0;34m"
+
+ERROR="${RED}[ERROR]"
+INFO="${GREEN}[INFO]"
+WARN="${YELLOW}[WARN]"
+
 function get_npu_model(){
     if command -v npu-smi &> /dev/null; then
         echo "Ascend$(npu-smi info -t board -i 0 -c 0 | awk '/Chip Name/ {print $NF}')"
@@ -22,12 +32,6 @@ function get_npu_model(){
     fi
 }
 
-RED='\033[0;31m'
-GREEN='\033[0;32m'
-YELLOW='\033[0;33m'
-BLUE='\033[0;34m'
-NC='\033[0m'
-
 SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
 CMAKE_SOURCE_DIR=$(realpath "$SCRIPT_DIR/..")
 BUILD_DIR="$CMAKE_SOURCE_DIR/build"
@@ -52,27 +56,34 @@ function show_help() {
     echo "  catlass_examples  Build Catlass examples"
     echo "  python_extension  Build Python extension"
     echo "  torch_library     Build Torch library"
-    echo "  <other>           Other CMake targets"
+    echo "  <other>           Other specific targets, e.g. 00_basic_matmul"
 }
 
 TARGET=""
 CMAKE_BUILD_TYPE="Release"
 declare -a CMAKE_OPTIONS=()
 CLEAN=false
-POST_BUILD_HINT=""
+POST_BUILD_INFO=""
+
+if [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
+    show_help
+    exit 0
+fi
+
+if [[ ! -v ASCEND_HOME_PATH ]]; then
+    echo "${ERROR}ASCEND_HOME_PATH environment variable is not set!${NC}"
+    echo "${ERROR}Please set ASCEND_HOME_PATH before running this script.${NC}"
+    exit 1
+fi
 
 while [[ $# -gt 0 ]]; do
     case "$1" in
-        -h|--help)
-            show_help
-            exit 0
-            ;;
         --clean)
             CLEAN=true
             ;;
         --debug)
             CMAKE_BUILD_TYPE="Debug"
-            echo -e "${YELLOW}Debug mode enabled${NC}"
+            echo -e "${WARN}Debug mode enabled"
             ;;
         --msdebug)
             CMAKE_OPTIONS+=("-DENABLE_MSDEBUG=True")
@@ -80,12 +91,13 @@ while [[ $# -gt 0 ]]; do
         --simulator)
             CMAKE_OPTIONS+=("-DENABLE_SIMULATOR=True")
             if NPU_MODEL=$(get_npu_model); then
-                echo -e "${YELLOW}No npu-smi detected, using default model for simulator: ${NPU_MODEL}${NC}"
+                echo -e "${WARN}No npu-smi detected, using default model for simulator: ${NPU_MODEL}${NC}"
             else
-                echo -e "${GREEN} Detect NPU_MODEL: ${NPU_MODEL}"
+                echo -e "${INFO}Detect NPU_MODEL: ${NPU_MODEL}${NC}"
             fi
             CMAKE_OPTIONS+=("-DSIMULATOR_NPU_MODEL=${NPU_MODEL}")
-            POST_BUILD_HINT="Please run \nexport LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/tools/simulator/${NPU_MODEL}/lib:\$LD_LIBRARY_PATH\nin your terminal before execute examples."            ;;
+            POST_BUILD_INFO="${INFO}Please run ${NC}\nexport LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/tools/simulator/${NPU_MODEL}/lib:\$LD_LIBRARY_PATH\n${GREEN}in your terminal before execute examples.${NC}"            
+            ;;
         --enable_msprof)
             CMAKE_OPTIONS+=("-DENABLE_MSPROF=True")
             ;;
@@ -96,7 +108,7 @@ while [[ $# -gt 0 ]]; do
             if [[ -z "$TARGET" ]]; then
                 TARGET="$1"
             else
-                echo -e "${RED}Error: Multiple targets specified${NC}" >&2
+                echo -e "${ERROR}Multiple targets specified${NC}" >&2
                 show_help
                 exit 1
             fi
@@ -106,13 +118,13 @@ while [[ $# -gt 0 ]]; do
 done
 
 if [[ "$CLEAN" == true ]]; then
-    echo -e "${BLUE}Cleaning build directories...${NC}"
+    echo -e "${INFO}Cleaning build directories...${NC}"
     rm -rf "$BUILD_DIR" "$OUTPUT_DIR"
-    echo -e "${GREEN}Clean complete${NC}"
+    echo -e "${INFO}Clean complete${NC}"
 fi
 
 if [[ -z "$TARGET" ]]; then
-    echo -e "${RED}Error: No target specified${NC}" >&2
+    echo -e "${ERROR}No target specified${NC}" >&2
     show_help
     exit 1
 fi
@@ -120,15 +132,15 @@ fi
 mkdir -p "$BUILD_DIR" "$OUTPUT_DIR"
 
 function build_python_extension() {
-    echo -e "${BLUE}Building Python extension...${NC}"
+    echo -e "${INFO}Building Python extension...${NC}"
     cd "$CMAKE_SOURCE_DIR/examples/python_extension" || exit 1
     rm -rf build dist ./*.egg-info
     python3 setup.py bdist_wheel --dist-dir "$OUTPUT_DIR/python_extension"
-    echo -e "${GREEN}Python extension built successfully${NC}"
+    echo -e "${INFO}Python extension built successfully${NC}"
 }
 
 function build_torch_library() {
-    echo -e "${BLUE}Building Torch library...${NC}"
+    echo -e "${INFO}Building Torch library...${NC}"
     cmake -B build \
         -DCMAKE_BUILD_TYPE="$CMAKE_BUILD_TYPE" \
         -DCMAKE_INSTALL_PREFIX="$OUTPUT_DIR/python_extension" \
@@ -137,7 +149,7 @@ function build_torch_library() {
         -DBUILD_TORCH_LIB=True
     cmake --build build --target catlass_torch -j
     cmake --install build --component catlass_torch
-    echo -e "${GREEN}Torch library built successfully${NC}"
+    echo -e "${INFO}Torch library built successfully${NC}"
 }
 
 # 执行构建
@@ -149,15 +161,15 @@ case "$TARGET" in
         build_torch_library
         ;;
     *)
-        echo -e "${BLUE}Building target: $TARGET...${NC}"
+        echo -e "${INFO}Building target: $TARGET...${NC}"
         cmake -S "$CMAKE_SOURCE_DIR" -B "$BUILD_DIR" \
             -DCMAKE_BUILD_TYPE="$CMAKE_BUILD_TYPE" \
             -DCMAKE_INSTALL_PREFIX="$OUTPUT_DIR" \
             "${CMAKE_OPTIONS[@]}"
         cmake --build "$BUILD_DIR" --target "$TARGET" -j
         cmake --install "$BUILD_DIR" --component "$TARGET"
-        echo -e "${GREEN}Target '$TARGET' built successfully${NC}"
+        echo -e "${INFO}Target '$TARGET' built successfully${NC}"
         ;;
 esac
 
-echo -e "$POST_BUILD_HINT"
\ No newline at end of file
+echo -e "$POST_BUILD_INFO"
\ No newline at end of file
-- 
Gitee


From 6fd7d40c1081df8d649ebed653355fd97c0b6c59 Mon Sep 17 00:00:00 2001
From: yuantao <taoyuan15@h-partners.com>
Date: Wed, 16 Jul 2025 08:43:03 +0000
Subject: [PATCH 6/7] support simulator

---
 scripts/build.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/build.sh b/scripts/build.sh
index e5aff6f2..27416213 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -71,8 +71,8 @@ if [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
 fi
 
 if [[ ! -v ASCEND_HOME_PATH ]]; then
-    echo "${ERROR}ASCEND_HOME_PATH environment variable is not set!${NC}"
-    echo "${ERROR}Please set ASCEND_HOME_PATH before running this script.${NC}"
+    echo -e "${ERROR}ASCEND_HOME_PATH environment variable is not set!${NC}"
+    echo -e "${ERROR}Please set ASCEND_HOME_PATH before running this script.${NC}"
     exit 1
 fi
 
-- 
Gitee


From 3cf755c9239ec09cc86b904974343252872e1414 Mon Sep 17 00:00:00 2001
From: yuantao <yuantao313@outlook.com>
Date: Wed, 16 Jul 2025 22:41:17 +0800
Subject: [PATCH 7/7] update enable to ascend_enable

---
 cmake/modules/CMakeASCENDInformation.cmake       | 2 +-
 cmake/modules/CMakeDetermineASCENDCompiler.cmake | 6 +++---
 scripts/build.sh                                 | 6 +++---
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/cmake/modules/CMakeASCENDInformation.cmake b/cmake/modules/CMakeASCENDInformation.cmake
index 1aa3853a..43473907 100644
--- a/cmake/modules/CMakeASCENDInformation.cmake
+++ b/cmake/modules/CMakeASCENDInformation.cmake
@@ -35,7 +35,7 @@ include(CMakeCommonLanguageInclude)
 
 set(CMAKE_ASCEND_FLAGS "-std=c++${CMAKE_CXX_STANDARD}")
 
-if(DEFINED ENABLE_MSDEBUG AND ENABLE_MSDEBUG)
+if(DEFINED ASCEND_ENABLE_MSDEBUG AND ASCEND_ENABLE_MSDEBUG)
     set(CMAKE_ASCEND_FLAGS_DEBUG "-O0 -g")
 else()
     set(CMAKE_ASCEND_FLAGS_DEBUG "-Xhost-start -O0 -g -Xhost-end -Xaicore-start -O2 -Xaicore-end")
diff --git a/cmake/modules/CMakeDetermineASCENDCompiler.cmake b/cmake/modules/CMakeDetermineASCENDCompiler.cmake
index 84c8a927..bfaf69c3 100644
--- a/cmake/modules/CMakeDetermineASCENDCompiler.cmake
+++ b/cmake/modules/CMakeDetermineASCENDCompiler.cmake
@@ -35,7 +35,7 @@ set(_CMAKE_ASCEND_HOST_IMPLICIT_LINK_LIBRARIES
     stdc++
 )
 
-if(DEFINED ENABLE_SIMULATOR AND ENABLE_SIMULATOR)
+if(DEFINED ASCEND_ENABLE_SIMULATOR AND ASCEND_ENABLE_SIMULATOR)
     if(NOT DEFINED SIMULATOR_NPU_MODEL)
         message(WARNING "Simulator mode is enabled but SIMULATOR_NPU_MODEL is not defined. Try get model from LD_LIBRARY_PATH.")
         set(LD_LIBRARY_PATH $ENV{LD_LIBRARY_PATH})
@@ -58,7 +58,7 @@ else()
         runtime)
 endif()
 
-if(DEFINED ENABLE_MSPROF AND ENABLE_MSPROF)
+if(DEFINED ASCEND_ENABLE_MSPROF AND ASCEND_ENABLE_MSPROF)
     list(APPEND _CMAKE_ASCEND_HOST_IMPLICIT_LINK_LIBRARIES profapi)
 endif()
 
@@ -68,7 +68,7 @@ set(_CMAKE_ASCEND_HOST_IMPLICIT_INCLUDE_DIRECTORIES
     ${_CMAKE_ASCEND_HOME_PATH}/include/experiment/msprof
 )
 
-if(NOT DEFINED ENABLE_ASCENDC OR ENABLE_ASCENDC)
+if(NOT DEFINED ASCEND_ENABLE_ASCENDC OR ASCEND_ENABLE_ASCENDC)
     list(APPEND _CMAKE_ASCEND_HOST_IMPLICIT_INCLUDE_DIRECTORIES
         ${_CMAKE_ASCEND_HOME_PATH}/compiler/tikcpp
         ${_CMAKE_ASCEND_HOME_PATH}/compiler/tikcpp/tikcfw
diff --git a/scripts/build.sh b/scripts/build.sh
index 27416213..f9d2a4f9 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -86,10 +86,10 @@ while [[ $# -gt 0 ]]; do
             echo -e "${WARN}Debug mode enabled"
             ;;
         --msdebug)
-            CMAKE_OPTIONS+=("-DENABLE_MSDEBUG=True")
+            CMAKE_OPTIONS+=("-DASCEND_ENABLE_MSDEBUG=True")
             ;;
         --simulator)
-            CMAKE_OPTIONS+=("-DENABLE_SIMULATOR=True")
+            CMAKE_OPTIONS+=("-DASCEND_ENABLE_SIMULATOR=True")
             if NPU_MODEL=$(get_npu_model); then
                 echo -e "${WARN}No npu-smi detected, using default model for simulator: ${NPU_MODEL}${NC}"
             else
@@ -99,7 +99,7 @@ while [[ $# -gt 0 ]]; do
             POST_BUILD_INFO="${INFO}Please run ${NC}\nexport LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/tools/simulator/${NPU_MODEL}/lib:\$LD_LIBRARY_PATH\n${GREEN}in your terminal before execute examples.${NC}"            
             ;;
         --enable_msprof)
-            CMAKE_OPTIONS+=("-DENABLE_MSPROF=True")
+            CMAKE_OPTIONS+=("-DASCEND_ENABLE_MSPROF=True")
             ;;
         -D*)
             CMAKE_OPTIONS+=("$1")
-- 
Gitee