Fix wrong indexing

EnzymeAD · Apr 3, 2024 · ea0f204 · ea0f204
1 parent 208d5cc
commit ea0f204
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/enzyme/benchmarks/ReverseMode/lstm/src/lib.rs b/enzyme/benchmarks/ReverseMode/lstm/src/lib.rs
@@ -90,7 +90,7 @@ fn lstm_predict(
         i += 2 * b;
     }
 
-    let xp = &s[i..];
+    let xp = &s[i-2*b..];
 
     for i in 0..b {
         x2[i] = xp[i] * w2[b + i] + w2[2 * b + i];