livoxG1: Add new OpenCl kernel to compact dgrams before collation

2025-11-09 02:39:09 -04:00
parent d2e2d9bc3b
commit a0a5aa49ad
3 changed files with 103 additions and 0 deletions
@@ -0,0 +1,83 @@
+__kernel void slotCompactor(
+	__global uchar* assembly,
+	uint numSlots,
+	uint slotStride,
+	uint slotSize,
+	uint firstSlotOffset,
+	uint nSucceeded)
+{
+	// Sequential processing: single work item processes all slots
+	// Compact non-dummy slots to the beginning (lowest addresses)
+	// Dummy slots will remain at the end (highest addresses)
+	// Optimizations:
+	// 1. For each dummy, find rightmost non-dummy and copy it there
+	// 2. Exit early once we've seen nSucceeded non-dummy slots
+	// 3. Exit early once we've moved nFailed dummy slots
+
+	uint nFailed = numSlots - nSucceeded;  // Calculate number of failed slots
+	uint nonDummiesSeen = 0;  // Track how many non-dummy slots we've seen
+	uint dummiesMoved = 0;     // Track how many dummy slots we've moved
+
+	// Process slots from beginning to end
+	for (uint i = 0; i < numSlots; ++i)
+	{
+		// Optimization 2: Exit early once we've seen nSucceeded non-dummy slots
+		if (nonDummiesSeen >= nSucceeded) { break; }
+		// Optimization 3: Exit early once we've moved nFailed dummy slots
+		if (dummiesMoved >= nFailed) { break; }
+
+		// Calculate slot address
+		__global uchar* slotAddr = assembly + firstSlotOffset
+			+ (i * slotStride);
+
+		// Check if slot is dummy: first 4 bytes should all be 0xFF
+		bool isDummy = (slotAddr[0] == 0xFF) && (slotAddr[1] == 0xFF)
+			&& (slotAddr[2] == 0xFF) && (slotAddr[3] == 0xFF);
+
+		if (isDummy)
+		{
+			// Optimization 1: Find rightmost non-dummy slot and copy it here
+			uint rightmostNonDummy = numSlots;
+			for (int j = (int)numSlots - 1; j > (int)i; --j)
+			{
+				__global uchar* checkSlotAddr = assembly + firstSlotOffset
+					+ (j * slotStride);
+
+				bool checkIsDummy = (checkSlotAddr[0] == 0xFF)
+					&& (checkSlotAddr[1] == 0xFF) && (checkSlotAddr[2] == 0xFF)
+					&& (checkSlotAddr[3] == 0xFF);
+
+				if (!checkIsDummy)
+				{
+					rightmostNonDummy = (uint)j;
+					break;
+				}
+			}
+
+			// If we found a non-dummy slot to the right, copy it here
+			if (rightmostNonDummy < numSlots)
+			{
+				__global uchar* srcAddr = assembly + firstSlotOffset
+					+ (rightmostNonDummy * slotStride);
+
+				// Copy slot data (byte-by-byte copy)
+				for (uint j = 0; j < slotSize; ++j) {
+					slotAddr[j] = srcAddr[j];
+				}
+
+				// Mark the source slot as dummy (move it to the end)
+				for (uint j = 0; j < 4; ++j) {
+					srcAddr[j] = 0xFF;
+				}
+
+				++dummiesMoved;
+				++nonDummiesSeen;  // We just moved a non-dummy to this position
+			}
+		}
+		else
+		{
+			// Slot is non-dummy - it's already in the right place
+			++nonDummiesSeen;
+		}
+	}
+}