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

Updated python tests

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-09 21:48:23 +00:00
parent d8041d696d
commit a828bb9f79
3 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
python/test/test_array_multiplication.py
View file

@ -14,7 +14,7 @@ def test_array_multiplication():
tensor_out = kp.Tensor([0, 0, 0])
# 3. Initialise the Kompute Tensors in the GPU
mgr.rebuild_tensors([tensor_in_a, tensor_in_b, tensor_out])
mgr.rebuild([tensor_in_a, tensor_in_b, tensor_out])
# 4. Define the multiplication shader code to run on the GPU
@ps.python2shader