kvcache: update tests

8b194b75 · Michael Yang · 3e8b8a19 · 8b194b75
Commit 8b194b75 authored Feb 26, 2025 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

kvcache/causal_test.go kvcache/causal_test.go +1 -1

No files found.
--- a/kvcache/causal_test.go
+++ b/kvcache/causal_test.go
@@ -342,7 +342,7 @@ func (c *testContext) FromIntSlice(s []int32, shape ...int) (ml.Tensor, error) {
 	return out, nil
 }

-func (c *testContext) Forward(ml.Tensor) {}
+func (c *testContext) Forward(...ml.Tensor) ml.Context { return c }

 func (c *testContext) Compute(...ml.Tensor) {}