From 66df76485b61b6021cca346f5508cfb4527d5b93 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B1=9F=E6=B3=A2?= Date: Thu, 19 Jun 2025 17:21:56 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=BF=E6=8D=A2=5F=5FCCE=5FKT=5FTEST=5F=5F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../AllGatherMatmulCustom/op_kernel/gather_mm.h | 2 +- .../op_kernel/matmul_reduce_scatter_custom_common.h | 2 +- .../op_kernel/matmul_all_reduce_custom_common.h | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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 9b662b32b..891f1082e 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 3d323216d..bb561cf03 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 95605f718..4dbf9e704 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 -- Gitee