From 79d518c50d9f70676cb556ce29bcd5227c29bf19 Mon Sep 17 00:00:00 2001
From: Sean Law <seanmylaw@gmail.com>
Date: Thu, 27 Feb 2025 10:05:56 -0500
Subject: [PATCH] Added njit numpy dot challenger

---
 sdp/challenger_sdp.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/sdp/challenger_sdp.py b/sdp/challenger_sdp.py
index ce60e8d..4833063 100644
--- a/sdp/challenger_sdp.py
+++ b/sdp/challenger_sdp.py
@@ -1,14 +1,18 @@
 import numpy as np
+from numba import njit
 
 
 def setup(Q, T):
+    sliding_dot_product(Q, T)
     return
 
 
+@njit(fastmath=True)
 def sliding_dot_product(Q, T):
-    m = len(Q)
+    m = Q.shape[0]
     l = T.shape[0] - m + 1
     out = np.empty(l)
     for i in range(l):
         out[i] = np.dot(Q, T[i : i + m])
+
     return out