diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index ccf15d0139ab4d86cbb3d296ac2206d0303a62a1..13319c6e38a0e43f11acaf4279fe1f69c7715c7b 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -13,7 +13,7 @@ cmake_minimum_required(VERSION 3.16.0)
 get_filename_component(ASCENDC_TESTS_DIR "${CMAKE_CURRENT_SOURCE_DIR}" ABSOLUTE)
 
 set(PRODUCT_TYPE_LIST ascend610 ascend310p ascend910B1_AIC ascend910B1_AIV ascend310B1)
-add_compile_definitions(__CCE_KT_TEST__=1 ASCENDC_CPU_DEBUG=1 ASCENDC_DUMP=0 LOG_CPP)
+add_compile_definitions(ASCENDC_CPU_DEBUG=1 ASCENDC_DUMP=0 LOG_CPP)
 
 if(BUILD_OPEN_PROJECT)
     include(cmake/config.cmake)
@@ -299,7 +299,6 @@ foreach(product_type ${PRODUCT_TYPE_LIST})
     endif()
 
     target_compile_definitions(ascendc_utest_${product_type} PRIVATE
-        __CCE_KT_TEST__=1
         ASCENDC_CPU_DEBUG=1
         ASCENDC_OOM=1
     )