new hls code with data reuse

triangle_counting_hls/triangle_counting_v2.cc

@@ -0,0 +1,73 @@
+#include <string.h>
+#include <stdio.h>
+
+#define BUF_SIZE 4096
+
+int triangle_counting(volatile unsigned int neighbor[9],
+                      volatile unsigned int offset[7],
+                      volatile unsigned int edge[18],
+                      unsigned int num_edges)
+{
+#pragma HLS INTERFACE m_axi port=neighbor depth = 32 offset=slave bundle=NEIGHBOR
+#pragma HLS INTERFACE m_axi port=offset depth = 32 offset=slave bundle=OFFSET
+#pragma HLS INTERFACE m_axi port=edge depth = 32 offset=slave bundle=EDGE
+#pragma HLS INTERFACE s_axilite register port=num_edges
+#pragma HLS INTERFACE s_axilite register port=return
+
+    unsigned int count = 0;
+    unsigned int i = 0;
+    unsigned int node_a, node_b;
+    unsigned int offset_a, offset_b;
+    unsigned int len_a, len_b;
+
+    unsigned int idx_a = 0;
+    unsigned int idx_b = 0;
+
+    unsigned int prev_node_a = 0;
+    unsigned int prev_node_b = 0;
+
+    unsigned int local_a[BUF_SIZE];
+    unsigned int local_b[BUF_SIZE];
+
+    for (i = 0; i < num_edges; i+=2)
+    {
+#pragma HLS pipeline
+        node_a = edge[i];
+        node_b = edge[i+1];
+
+        if (node_a != prev_node_a)
+        {
+            offset_a = offset[node_a];
+            len_a = offset[node_a + 1] - offset_a;
+            memcpy(local_a, (const unsigned int*)&(neighbor[offset_a]), len_a*sizeof(unsigned int));
+            prev_node_a = node_a; 
+        }
+
+        if (node_b != prev_node_b)
+        {
+            offset_b = offset[node_b];
+            len_b = offset[node_b + 1] - offset_b;
+            memcpy(local_b, (const unsigned int*)&(neighbor[offset_b]), len_b*sizeof(unsigned int));
+            prev_node_b = node_b;
+        }
+
+        idx_a = 0;
+        idx_b = 0;
+
+        while (idx_a < len_a && idx_b < len_b)
+        {
+    #pragma HLS pipeline
+            if (local_a[idx_a] < local_b[idx_b])
+                idx_a++;
+            else if (local_a[idx_a] > local_b[idx_b])
+                idx_b++;
+            else
+            {
+                count++;
+                idx_a++;
+                idx_b++;
+            }
+        }
+    }
+    return count;
+}