kvcache: fix tests by adding AvgPool2D stub

c6b6938b · jmorganca · Michael Yang · fb4664fc · c6b6938b
Commit c6b6938b authored Mar 11, 2025 by jmorganca Committed by Michael Yang Mar 11, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

kvcache/causal_test.go kvcache/causal_test.go +4 -0

No files found.
--- a/kvcache/causal_test.go
+++ b/kvcache/causal_test.go
@@ -445,6 +445,10 @@ func (t *testTensor) AvgPool1D(ctx ml.Context, k, s, p int) ml.Tensor {
 	panic("not implemented")
 }
+func (t *testTensor) AvgPool2D(ctx ml.Context, k, s int, p float32) ml.Tensor {
+	panic("not implemented")
+}
 func (t *testTensor) Conv2D(ctx ml.Context, weight ml.Tensor, s0, s1, p0, p1, d0, d1 int) ml.Tensor {
 	panic("not implemented")
 }