minor

microsoft · Dec 28, 2023 · 6033344 · 6033344
1 parent 8c8c8fe
commit 6033344
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git a/DPDPU/OffloadEngine/DPU/Include/UDF.h b/DPDPU/OffloadEngine/DPU/Include/UDF.h
@@ -21,8 +21,8 @@ typedef void (*OffloadPred)(
     CacheTableT* CacheTable,
     RequestDescriptorT* ReqsForHost,
     RequestDescriptorT* ReqsForDPU,
-    uint8_t* NumReqsForHost,
-    uint8_t* NumReqsForDPU
+    uint16_t* NumReqsForHost,
+    uint16_t* NumReqsForDPU
 );
 
 //
@@ -81,4 +81,4 @@ GetOffloadFunc(
 void
 UnloadLibrary(
     void *Lib
-);
+);
diff --git a/DPDPU/OffloadEngine/UDFSamples/OffloadPred.c b/DPDPU/OffloadEngine/UDFSamples/OffloadPred.c
@@ -16,8 +16,8 @@ void OffloadPred(
     CacheTableT* CacheTable,
     RequestDescriptorT* ReqsForHost,
     RequestDescriptorT* ReqsForDPU,
-    uint8_t* NumReqsForHost,
-    uint8_t* NumReqsForDPU
+    uint16_t* NumReqsForHost,
+    uint16_t* NumReqsForDPU
 ) {
     const uint16_t reqSize = sizeof(struct MessageHeader);
 

diff --git a/DPDPU/OffloadEngine/UDFSamples/OffloadPredDisaggApp.c b/DPDPU/OffloadEngine/UDFSamples/OffloadPredDisaggApp.c
@@ -16,8 +16,8 @@ void OffloadPred(
     CacheTableT* CacheTable,
     RequestDescriptorT* ReqsForHost,
     RequestDescriptorT* ReqsForDPU,
-    uint8_t* NumReqsForHost,
-    uint8_t* NumReqsForDPU
+    uint16_t* NumReqsForHost,
+    uint16_t* NumReqsForDPU
 ) {
     const uint16_t reqSize = sizeof(struct MessageHeader);
     const int numReqsForDPU = Bytes / reqSize;
@@ -29,4 +29,4 @@ void OffloadPred(
         ReqsForDPU[i].Offset = i * reqSize;
         ReqsForDPU[i].Bytes = reqSize;    
     }
-}
+}
diff --git a/DPDPU/OffloadEngine/UDFSamples/OffloadPredExample.c b/DPDPU/OffloadEngine/UDFSamples/OffloadPredExample.c
@@ -15,8 +15,8 @@ void OffloadPred(
     CacheTableT* CacheTable,
     RequestDescriptorT* ReqsForHost,
     RequestDescriptorT* ReqsForDPU,
-    uint8_t* NumReqsForHost,
-    uint8_t* NumReqsForDPU
+    uint16_t* NumReqsForHost,
+    uint16_t* NumReqsForDPU
 ) {
     *NumReqsForHost = 1;
     *NumReqsForDPU = 0;
@@ -30,8 +30,8 @@ void OffloadPred(
     CacheTableT* CacheTable,
     RequestDescriptorT* ReqsForHost,
     RequestDescriptorT* ReqsForDPU,
-    uint8_t* NumReqsForHost,
-    uint8_t* NumReqsForDPU
+    uint16_t* NumReqsForHost,
+    uint16_t* NumReqsForDPU
 ) {
     *NumReqsForHost = 0;
     *NumReqsForDPU = 1;