diff --git a/operator/ascendc/4_best_practices/21_all_gather_matmul_custom/AllGatherMatmulCustom/op_kernel/gather_mm.h b/operator/ascendc/4_best_practices/21_all_gather_matmul_custom/AllGatherMatmulCustom/op_kernel/gather_mm.h
index 9b662b32bcaaa35440a68dcd78f3261928fbae7c..891f1082eff3526c5ccf458468dc3f2c842a2332 100644
--- a/operator/ascendc/4_best_practices/21_all_gather_matmul_custom/AllGatherMatmulCustom/op_kernel/gather_mm.h
+++ b/operator/ascendc/4_best_practices/21_all_gather_matmul_custom/AllGatherMatmulCustom/op_kernel/gather_mm.h
@@ -11,7 +11,7 @@
 #ifndef MC2_GATHER_MM_H
 #define MC2_GATHER_MM_H
 
-#if defined(__CCE_KT_TEST__)
+#if defined ASCENDC_CPU_DEBUG
 #define SET_G_CORE_TYPE_IS_AIV thread_local int g_coreType = 2
 #define SET_G_CORE_TYPE_IS_AIC thread_local int g_coreType = 1
 #define DTYPE_X1 half
diff --git a/operator/ascendc/4_best_practices/22_matmul_reduce_scatter_custom/MatmulReduceScatterCustom/op_kernel/matmul_reduce_scatter_custom_common.h b/operator/ascendc/4_best_practices/22_matmul_reduce_scatter_custom/MatmulReduceScatterCustom/op_kernel/matmul_reduce_scatter_custom_common.h
index 3d323216ded5039c92a9d0c4a90bad0762dca9d2..bb561cf0301afef18e31b1800ad2321748389ab0 100644
--- a/operator/ascendc/4_best_practices/22_matmul_reduce_scatter_custom/MatmulReduceScatterCustom/op_kernel/matmul_reduce_scatter_custom_common.h
+++ b/operator/ascendc/4_best_practices/22_matmul_reduce_scatter_custom/MatmulReduceScatterCustom/op_kernel/matmul_reduce_scatter_custom_common.h
@@ -11,7 +11,7 @@
 #ifndef MC2_ALLREDUCE_COMM_H
 #define MC2_ALLREDUCE_COMM_H
 
-#if defined(__CCE_KT_TEST__)
+#if defined ASCENDC_CPU_DEBUG
 #define SET_G_CORE_TYPE_IS_AIV thread_local int g_coreType = 2
 #define SET_G_CORE_TYPE_IS_AIC thread_local int g_coreType = 1
 #define DTYPE_X1 half
diff --git a/operator/ascendc/4_best_practices/23_matmul_all_reduce_custom/MatmulAllReduceCustom/op_kernel/matmul_all_reduce_custom_common.h b/operator/ascendc/4_best_practices/23_matmul_all_reduce_custom/MatmulAllReduceCustom/op_kernel/matmul_all_reduce_custom_common.h
index 95605f7185adb7b02408d80589db92cef6fed9d0..4dbf9e7045abcdda1eb3badbe56b576596876486 100644
--- a/operator/ascendc/4_best_practices/23_matmul_all_reduce_custom/MatmulAllReduceCustom/op_kernel/matmul_all_reduce_custom_common.h
+++ b/operator/ascendc/4_best_practices/23_matmul_all_reduce_custom/MatmulAllReduceCustom/op_kernel/matmul_all_reduce_custom_common.h
@@ -11,7 +11,7 @@
 #ifndef MC2_ALLREDUCE_COMM_H
 #define MC2_ALLREDUCE_COMM_H
 
-#if defined(__CCE_KT_TEST__)
+#if defined ASCENDC_CPU_DEBUG
 #define SET_G_CORE_TYPE_IS_AIV thread_local int g_coreType = 2
 #define SET_G_CORE_TYPE_IS_AIC thread_local int g_coreType = 1
 #define DTYPE_X1 half