diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 55687963ad0acb5d652c06e9ad90559d3e0da7ef..dfbdcdad304fbcbc358c4ec64121e013c1c5d367 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}