From bc03e9051210196fdedc9bedeac957809bd662fe Mon Sep 17 00:00:00 2001
From: pc9ha2xvl <2119910569@qq.com>
Date: Sat, 30 Nov 2024 00:50:32 +0800
Subject: [PATCH] ADD file via upload

---
 step5.c | 70 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 step5.c

diff --git a/step5.c b/step5.c
new file mode 100644
index 0000000..1264540
--- /dev/null
+++ b/step5.c
@@ -0,0 +1,70 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <time.h>
+
+#define MAX 1000 
+
+
+void sparse_matmul_coo(float* A_values, int* A_rowIndex, int* A_colIndex, int A_nonZeroCount,
+                       float* B_values, int* B_rowIndex, int* B_colIndex, int B_nonZeroCount,
+                       float* C_values, int* C_rowIndex, int* C_colIndex, int* C_nonZeroCount) {
+    int currentIndex = 0;
+    int i;
+    for (i = 0; i < A_nonZeroCount; i++) {
+        int rowA = A_rowIndex[i];
+        int colA = A_colIndex[i];
+        float valueA = A_values[i];
+        int j;
+        for (j = 0; j < B_nonZeroCount; j++) {
+            int rowB = B_rowIndex[j];
+            int colB = B_colIndex[j];
+            float valueB = B_values[j];
+            if (colA == rowB) { 
+                float product = valueA * valueB;
+                int found = 0;
+                int k;
+                for ( k = 0; k < currentIndex; k++) {
+                    if (C_rowIndex[k] == rowA && C_colIndex[k] == colB) {
+                        C_values[k] += product; 
+                        found = 1;
+                        break;
+                    }
+                }
+                if (!found) { 
+                    C_values[currentIndex] = product;
+                    C_rowIndex[currentIndex] = rowA;
+                    C_colIndex[currentIndex] = colB;
+                    currentIndex++;
+                }
+            }
+        }
+    }
+    *C_nonZeroCount = currentIndex; 
+}
+
+int main() {
+    float A_values[] = {1, 2, 3, 4, 5};
+    int A_rowIndex[] = {0, 0, 1, 2, 2};
+    int A_colIndex[] = {0, 2, 1, 0, 2};
+    int A_nonZeroCount = 5;
+
+    float B_values[] = {6, 8, 7, 9};
+    int B_rowIndex[] = {0, 2, 1, 2};
+    int B_colIndex[] = {0, 0, 1, 2};
+    int B_nonZeroCount = 4;
+
+    float C_values[MAX];
+    int C_rowIndex[MAX];
+    int C_colIndex[MAX];
+    int C_nonZeroCount = 0;
+
+    clock_t start = clock();
+    sparse_matmul_coo(A_values, A_rowIndex, A_colIndex, A_nonZeroCount,
+                      B_values, B_rowIndex, B_colIndex, B_nonZeroCount,
+                      C_values, C_rowIndex, C_colIndex, &C_nonZeroCount);
+    clock_t end = clock();
+    double time_taken = (double)(end - start) / CLOCKS_PER_SEC;
+    printf(" %f s\n", time_taken);
+
+    return 0;
+}