kvcache: update tests

This commit is contained in:
Michael Yang 2025-02-26 12:16:59 -08:00
parent 3e8b8a1933
commit 8b194b7520

View File

@ -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) {}