feat: add benchmarking for Viterbi decoder for terminated codes

rwnobrega · Dec 12, 2024 · c3d642e · c3d642e
1 parent 3e3954c
commit c3d642e
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/benchmarks/error_control_convolutional/bench_convolutional.py b/benchmarks/error_control_convolutional/bench_convolutional.py
@@ -0,0 +1,33 @@
+import numpy as np
+
+import komm
+
+
+def bench_decode_convolutional_mu_2(benchmark):
+    convolutional_code = komm.ConvolutionalCode(
+        feedforward_polynomials=[[0o7, 0o5]],
+    )
+    code = komm.TerminatedConvolutionalCode(
+        convolutional_code,
+        num_blocks=200,
+        mode="zero-termination",
+    )
+    decoder = komm.ViterbiDecoder(code)
+    num_frames = 10
+    r = np.random.randint(0, 2, size=(num_frames, code.length))
+    benchmark(decoder, r)
+
+
+def bench_decode_convolutional_mu_6(benchmark):
+    convolutional_code = komm.ConvolutionalCode(
+        feedforward_polynomials=[[0o117, 0o155]]
+    )
+    code = komm.TerminatedConvolutionalCode(
+        convolutional_code,
+        num_blocks=200,
+        mode="zero-termination",
+    )
+    decoder = komm.ViterbiDecoder(code)
+    num_frames = 10
+    r = np.random.randint(0, 2, size=(num_frames, code.length))
+    benchmark(decoder, r)