From c6b6938b3a5d0aef109464e8d42c7849d72bacce Mon Sep 17 00:00:00 2001
From: jmorganca <jmorganca@gmail.com>
Date: Tue, 11 Mar 2025 18:57:48 +0100
Subject: [PATCH] kvcache: fix tests by adding AvgPool2D stub

---
 kvcache/causal_test.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/kvcache/causal_test.go b/kvcache/causal_test.go
index ed23cad6a..56d85ceb6 100644
--- 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")
 }