diff --git a/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h b/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h index ebb6b0ff43..1dc24975ad 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h +++ b/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h @@ -59,8 +59,8 @@ constexpr auto kCustRunApi = "RunCpuKernel"; constexpr auto kDropout2D = "Dropout2D"; constexpr auto kDropout3D = "Dropout3D"; const std::set kCustAiCpuKernelOps{kIdentity}; -const std::set kCacheKernelOps{kUpdateCache, kCacheSwapTable, kSubAndFilter, kPadAndShift, kDropout3D, - kDropout2D, kStackInit, kStackPush, kStackPop, kStackDestroy}; +const std::set kCacheKernelOps{kUpdateCache, kCacheSwapTable, kSubAndFilter, + kPadAndShift, kDropout3D, kDropout2D}; struct AicpuParamHead { uint32_t length; // Total length: include cunstom message