Skip to content

GitLab

Commit 8b194b75 authored by Michael Yang's avatar Michael Yang
Browse files

kvcache: update tests

parent 3e8b8a19
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
kvcache/causal_test.go
View file @ 8b194b75
...@@ -342,7 +342,7 @@ func (c *testContext) FromIntSlice(s []int32, shape ...int) (ml.Tensor, error) { ...@@ -342,7 +342,7 @@ func (c *testContext) FromIntSlice(s []int32, shape ...int) (ml.Tensor, error) {
return out, nil 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) {} func (c *testContext) Compute(...ml.Tensor) {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment