From 08cae2a7b8662e8f5df3c0b382a04868e9d2f069 Mon Sep 17 00:00:00 2001
From: zhangyike <zhangyike5@huawei.com>
Date: Fri, 22 Aug 2025 09:49:23 +0000
Subject: [PATCH] add __NPU_ARCH__ for utest/CMakeLists.txt.

Signed-off-by: zhangyike <zhangyike5@huawei.com>
---
 tests/CMakeLists.txt | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 55687963..dfbdcdad 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -245,6 +245,7 @@ foreach(product_type ${PRODUCT_TYPE_LIST})
             __CCE_AICORE__=200
             __DAV_M200__
             __UT_CHIP_VER__=610
+            __NPU_ARCH__=2002
         )
         target_link_directories(ascendc_utest_${product_type} PRIVATE
             ${ASCEND_DEP_LIB_DIR}
@@ -257,6 +258,7 @@ foreach(product_type ${PRODUCT_TYPE_LIST})
             __CCE_AICORE__=200
             __DAV_M200__
             __UT_CHIP_VER__=710
+            __NPU_ARCH__=2002
         )
         target_link_directories(ascendc_utest_${product_type} PRIVATE
             ${ASCEND_DEP_LIB_DIR}
@@ -269,6 +271,7 @@ foreach(product_type ${PRODUCT_TYPE_LIST})
             __CCE_AICORE__=220
             __DAV_C220__
             __DAV_C220_CUBE__
+            __NPU_ARCH__=2201
         )
         target_link_directories(ascendc_utest_${product_type} PRIVATE
             ${ASCEND_DEP_LIB_DIR}
@@ -281,6 +284,7 @@ foreach(product_type ${PRODUCT_TYPE_LIST})
             __CCE_AICORE__=220
             __DAV_C220__
             __DAV_C220_VEC__
+            __NPU_ARCH__=2201
         )
         target_link_directories(ascendc_utest_${product_type} PRIVATE
             ${ASCEND_DEP_LIB_DIR}
@@ -293,6 +297,7 @@ foreach(product_type ${PRODUCT_TYPE_LIST})
             __CCE_AICORE__=300
             __DAV_M300__
             __UT_CHIP_VER__=300
+            __NPU_ARCH__=3002
         )
         target_link_directories(ascendc_utest_${product_type} PRIVATE
             ${ASCEND_DEP_LIB_DIR}
-- 
Gitee