kvcache: Set context for shift offsets

f52b2615 · Jesse Gross · Jesse Gross · 25f9b152 · f52b2615
Commit f52b2615 authored Mar 07, 2025 by Jesse Gross Committed by Jesse Gross Mar 07, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

kvcache/causal.go kvcache/causal.go +1 -1

No files found.
--- a/kvcache/causal.go
+++ b/kvcache/causal.go
@@ -545,7 +545,7 @@ func (c *Causal) shift(seq int, beginIndex, offset int32) error {
 		}
 	}

-	kShift, err := ctx.FromIntSlice(offsets, len(offsets))
+	kShift, err := ctx.Input().FromIntSlice(offsets, len(offsets))
 	if err != nil {
 		return err
 	}