From 95fbf1da12bd7bbc984cf902ff5fe87336b7e5f8 Mon Sep 17 00:00:00 2001 From: Patrick Devine Date: Tue, 18 Feb 2025 13:02:44 -0800 Subject: [PATCH] fix causal test --- kvcache/causal_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kvcache/causal_test.go b/kvcache/causal_test.go index 874e47433..dca2b83d5 100644 --- a/kvcache/causal_test.go +++ b/kvcache/causal_test.go @@ -434,7 +434,7 @@ func (t *testTensor) Conv2D(ctx ml.Context, weight ml.Tensor, s0, s1, p0, p1, d0 panic("not implemented") } -func (t *testTensor) RoPE(ctx ml.Context, positionIDs, ropeFactors ml.Tensor, dim uint32, base, scale float32) ml.Tensor { +func (t *testTensor) RoPE(ctx ml.Context, positionIDs, ropeFactors ml.Tensor, dim, ropeType uint32, base, scale float32) ml.Tensor { panic("not implemented") }