From 59c3e5126f83f5fd7f1d2feff76f2809a561b113 Mon Sep 17 00:00:00 2001
From: miha-q <>
Date: Thu, 7 Mar 2024 18:21:58 -0500
Subject: [PATCH] Thu Mar  7 06:21:58 PM EST 2024

---
 src/QAnsel.c  |  4 ++++
 src/complex.c | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)

diff --git a/src/QAnsel.c b/src/QAnsel.c
index ac7e82b..7b29e1a 100644
--- a/src/QAnsel.c
+++ b/src/QAnsel.c
@@ -242,6 +242,10 @@ void qansel_instruction(cpx_mtx_t* stateVector, unsigned char qubitCount, QInstr
 		us1 = get_time();
 		cpx_mtx_knk_threads(tmp.ptr, filter.ptr, gate.ptr, filter.rows, filter.cols, gate.rows, gate.cols);
 		us2 = get_time();
+		printf("\tThreads2x2: %lu\n", us2 - us1);
+		us1 = get_time();
+		cpx_mtx_knk_threads_2x2(tmp.ptr, filter.ptr, gate.ptr, filter.rows, filter.cols, gate.rows, gate.cols);
+		us2 = get_time();
 		printf("\tThreads: %lu\n", us2 - us1);
 		us1 = get_time();
 		cpx_mtx_knk(tmp.ptr, filter.ptr, gate.ptr, filter.rows, filter.cols, gate.rows, gate.cols);
diff --git a/src/complex.c b/src/complex.c
index 2f6b8da..a9ca17c 100644
--- a/src/complex.c
+++ b/src/complex.c
@@ -220,6 +220,56 @@ void cpx_mtx_knk_threads(float* ptrR, float* ptrA, float* ptrB, int rowsA, int c
     }
 }
 
+void* cpx_mtx_knk_threads_2x2_run(void *context)
+{
+	cpx_thread_context* ctx = (cpx_thread_context*)context;
+	int rowsR = (ctx->rowsA) * (ctx->rowsB);
+	int colsR = (ctx->colsA) * (ctx->colsB);
+    for (int i = 0; i < (ctx->delimeterCount); i++)
+    {
+		kernel_knk_2x2(ctx->ptrR, ctx->ptrA, ctx->rowsA, ctx->colsA, ctx->ptrB[0], ctx->ptrB[1], ctx->ptrB[2], ctx->ptrB[3], ctx->ptrB[4], ctx->ptrB[5], ctx->ptrB[6], ctx->ptrB[7], i + (ctx->delimeterStart));
+    }
+}
+
+void cpx_mtx_knk_threads_2x2(float* ptrR, float* ptrA, float* ptrB, int rowsA, int colsA, int rowsB, int colsB)
+{
+	int delimeter = (rowsA * rowsB) / 2;
+	int cores = get_core_count();
+	int threadCount = cores;
+	if (threadCount > delimeter) threadCount = delimeter;
+	int delimetersPerThread = delimeter / threadCount;
+	int leftOvers = delimeter % threadCount;
+
+	cpx_thread_context ctx = {ptrR, ptrA, ptrB, rowsA, colsA, rowsB, colsB, 0, 0};
+	cpx_thread_context ctxs[threadCount];
+    pthread_t threads[threadCount];
+	for (int i = 0; i < threadCount; i++)
+	{
+		ctxs[i].ptrR = ctx.ptrR;
+		ctxs[i].ptrA = ctx.ptrA;
+		ctxs[i].ptrB = ctx.ptrB;
+		ctxs[i].rowsA = ctx.rowsA;
+		ctxs[i].colsA = ctx.colsA;
+		ctxs[i].rowsB = ctx.rowsB;
+		ctxs[i].colsB = ctx.colsB;
+		ctxs[i].delimeterStart = i * delimetersPerThread;
+		ctxs[i].delimeterCount = delimetersPerThread + ((i == threadCount - 1) ? leftOvers : 0);
+
+        if (pthread_create(&(threads[i]), NULL, &cpx_mtx_knk_threads_2x2_run, (void*)&(ctxs[i])))
+        {
+            fprintf(stderr, "QAnsel: Thread error. (1)\n");
+            exit(1);
+        }
+	}
+    for (uint32_t i = 0; i < threadCount; i++)
+    {
+        if (pthread_join(threads[i], NULL))
+        {
+            fprintf(stderr, "QAnsel: Thread error. (2)\n");
+        }
+    }
+}
+
 void* cpx_mtx_dot_threads_run(void *context)
 {
 	cpx_thread_context* ctx = (cpx_thread_context*)context;
-- 
2.39.5