From c7e1c5c965de9f6e5ce42a39165679d3075d4640 Mon Sep 17 00:00:00 2001 From: yanzhenxiang2020 Date: Mon, 15 Mar 2021 17:54:17 +0800 Subject: [PATCH] mv stack pop/pop to ms_kernel --- mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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