thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

added tests

Browse source
This commit is contained in:
--- 2020-12-27 13:26:41 +01:00
parent 46278eb0a9
commit 695fb08c80
3 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
python/test/test_array_multiplication.py
View file

@ -1,5 +1,6 @@
import pyshader as ps
import kp
import numpy as np
def test_array_multiplication():
@ -33,3 +34,4 @@ def test_array_multiplication():
mgr.eval_tensor_sync_local_def([tensor_out])
assert tensor_out.data() == [2.0, 4.0, 6.0]
assert np.all(tensor_out.numpy() == [2.0, 4.0, 6.0])