Browse Source

mv stack pop/pop to ms_kernel

tags/v1.2.0-rc1
yanzhenxiang2020 4 years ago
parent
commit
c7e1c5c965
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h

+ 2
- 2
mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h View File

@@ -59,8 +59,8 @@ constexpr auto kCustRunApi = "RunCpuKernel";
constexpr auto kDropout2D = "Dropout2D";
constexpr auto kDropout3D = "Dropout3D";
const std::set<std::string> kCustAiCpuKernelOps{kIdentity};
const std::set<std::string> kCacheKernelOps{kUpdateCache, kCacheSwapTable, kSubAndFilter, kPadAndShift, kDropout3D,
kDropout2D, kStackInit, kStackPush, kStackPop, kStackDestroy};
const std::set<std::string> kCacheKernelOps{kUpdateCache, kCacheSwapTable, kSubAndFilter,
kPadAndShift, kDropout3D, kDropout2D};

struct AicpuParamHead {
uint32_t length; // Total length: include cunstom message


Loading…
Cancel
Save