update AicpuParamHead for aicpu

5 years ago · f07f6655ee
parent 60de9089ba
commit f07f6655ee
1 changed files with 4 additions and 2 deletions
--- a/mindspore/ccsrc/kernel/aicpu/aicpu_util.h
+++ b/mindspore/ccsrc/kernel/aicpu/aicpu_util.h
@ -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 {