Browse Source

update AicpuParamHead for aicpu

tags/v0.5.0-beta
yanzhenxiang2020 5 years ago
parent
commit
f07f6655ee
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