Browse Source

!2290 update AicpuParamHead

Merge pull request !2290 from yanzhenxiang2020/aicpu_param_open
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
5d3792c273
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      mindspore/ccsrc/kernel/aicpu/aicpu_util.h

+ 4
- 2
mindspore/ccsrc/kernel/aicpu/aicpu_util.h View File

@@ -43,8 +43,10 @@ constexpr auto kTopK = "TopK";
constexpr auto kTopKV2 = "TopKV2";

struct AicpuParamHead {
uint32_t length; // Total length: include cunstom message
uint32_t ioAddrNum; // Input and output address number
uint32_t length; // Total length: include cunstom message
uint32_t ioAddrNum; // Input and output address number
uint32_t extInfoLength; // extInfo struct Length
uint64_t extInfoAddr; // extInfo address
} __attribute__((packed));

class AicpuOpUtil {


Loading…
Cancel
Save