1 files changed, 63 insertions, 0 deletions
diff --git a/tester/simd.c b/tester/simd.c
new file mode 100644
index 0000000..bea2f06
--- /dev/null
+++ b/tester/simd.c
@@ -0,0 +1,63 @@
+#include <stdio.h>
+
+#include "tests.h"
+
+static const double R2 = 0.7071067811865476;
+
+void run_simd_tests(void)
+{
+	psi_test_section("SIMD single-qubit kernels");
+
+	printf("  detected: %s\n", psi_simd_name(psi_detect_simd()));
+
+	struct PsiComplex h[2][2] = {
+		{ psi_new_complex(R2, 0.0), psi_new_complex(R2, 0.0) },
+		{ psi_new_complex(R2, 0.0), psi_new_complex(-R2, 0.0) },
+	};
+
+	struct PsiComplex two[4] = {
+		psi_new_complex(1.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+	};
+	psi_apply_single_qubit_gate_simd(two, h, 0, 2);
+	struct PsiComplex two_exp[] = {
+		psi_new_complex(R2, 0.0),
+		psi_new_complex(0.0, 0.0),
+		psi_new_complex(R2, 0.0),
+		psi_new_complex(0.0, 0.0),
+	};
+	psi_test_check(psi_amps_match(two, two_exp, 4), "SIMD H on q0 of |00>");
+
+	struct PsiComplex three[8];
+	for (size_t i = 0; i < 8; i++)
+		three[i] = psi_new_complex(0.0, 0.0);
+	three[0] = psi_new_complex(1.0, 0.0);
+	psi_apply_single_qubit_gate_simd(three, h, 0, 3);
+	struct PsiComplex three_exp[8];
+	for (size_t i = 0; i < 8; i++)
+		three_exp[i] = psi_new_complex(0.0, 0.0);
+	three_exp[0] = psi_new_complex(R2, 0.0);
+	three_exp[4] = psi_new_complex(R2, 0.0);
+	psi_test_check(psi_amps_match(three, three_exp, 8), "SIMD H on q0 of |000> (chunked)");
+
+	struct PsiComplex x[2][2] = {
+		{ psi_new_complex(0.0, 0.0), psi_new_complex(1.0, 0.0) },
+		{ psi_new_complex(1.0, 0.0), psi_new_complex(0.0, 0.0) },
+	};
+	struct PsiComplex flip[4] = {
+		psi_new_complex(1.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+	};
+	psi_apply_single_qubit_gate_simd(flip, x, 1, 2);
+	struct PsiComplex flip_exp[] = {
+		psi_new_complex(0.0, 0.0),
+		psi_new_complex(1.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+		psi_new_complex(0.0, 0.0),
+	};
+	psi_test_check(psi_amps_match(flip, flip_exp, 4), "SIMD X on q1 of |00> = |01>");
+}