From db3a119edce2dae02f3b4b8ea06ca81a36cc51a4 Mon Sep 17 00:00:00 2001 From: Belief Date: Thu, 11 Jul 2024 21:59:24 +0800 Subject: [PATCH] change CCE_KT_TEST into ASCENDC_CPU_DEBUG --- .../kernel_launch_method_by_direct/layernorm_custom.cpp | 2 +- .../layernorm/kernel_launch_method_by_direct/main.cpp | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/examples/normalization/layernorm/kernel_launch_method_by_direct/layernorm_custom.cpp b/examples/normalization/layernorm/kernel_launch_method_by_direct/layernorm_custom.cpp index 9d8604ee..87164a41 100644 --- a/examples/normalization/layernorm/kernel_launch_method_by_direct/layernorm_custom.cpp +++ b/examples/normalization/layernorm/kernel_launch_method_by_direct/layernorm_custom.cpp @@ -35,7 +35,7 @@ extern "C" __global__ __aicore__ void LayernormCustom(GM_ADDR inputXGm, GM_ADDR op.Process(); } -#ifndef __CCE_KT_TEST__ +#ifndef ASCENDC_CPU_DEBUG // call of kernel function void LayernormCustomDo(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *inputXGm, uint8_t *gammaGm, uint8_t *betaGm, uint8_t *outputGm, uint8_t *outputMeanGm, uint8_t *outputVarianceGm, uint8_t *workspace, diff --git a/examples/normalization/layernorm/kernel_launch_method_by_direct/main.cpp b/examples/normalization/layernorm/kernel_launch_method_by_direct/main.cpp index 3c65fd3f..c1e361f3 100644 --- a/examples/normalization/layernorm/kernel_launch_method_by_direct/main.cpp +++ b/examples/normalization/layernorm/kernel_launch_method_by_direct/main.cpp @@ -9,7 +9,7 @@ */ #include "../../../common/data_utils.h" -#ifndef __CCE_KT_TEST__ +#ifndef ASCENDC_CPU_DEBUG #include "acl/acl.h" extern void LayernormCustomDo(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *inputXGm, uint8_t *gammaGm, uint8_t *betaGm, uint8_t *outputGm, uint8_t *outputMeanGm, uint8_t *outputVarianceGm, uint8_t *workspace, @@ -32,7 +32,7 @@ extern uint8_t *GenerateTiling(uint32_t bLength, uint32_t sLength, uint32_t hLen static bool CompareResult(const void *outputData, int64_t outSize, std::string goldenName) { void *goldenData; -#ifdef __CCE_KT_TEST__ +#ifdef ASCENDC_CPU_DEBUG goldenData = (uint8_t *)AscendC::GmAlloc(outSize); #else CHECK_ACL(aclrtMallocHost((void **)(&goldenData), outSize)); @@ -59,7 +59,7 @@ static bool CompareResult(const void *outputData, int64_t outSize, std::string g wrongNum++; } } -#ifdef __CCE_KT_TEST__ +#ifdef ASCENDC_CPU_DEBUG AscendC::GmFree((void *)goldenData); #else CHECK_ACL(aclrtFreeHost(goldenData)); @@ -84,7 +84,7 @@ int32_t main(int32_t argc, char *argv[]) size_t outputSize_variance = BLENGTH * SLENGTH * sizeof(float); size_t tilingFileSize = TILINGDATA_SIZE * sizeof(uint32_t); -#ifdef __CCE_KT_TEST__ +#ifdef ASCENDC_CPU_DEBUG uint8_t *inputX = (uint8_t *)AscendC::GmAlloc(inputSize_inputX); uint8_t *gamma = (uint8_t *)AscendC::GmAlloc(inputSize_gamma); uint8_t *beta = (uint8_t *)AscendC::GmAlloc(inputSize_beta); -- Gitee