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

Made python tests location agnostic whre they get called

Browse source
This commit is contained in:
Alejandro Saucedo 2020-11-09 19:58:27 +00:00
parent b687e982d8
commit 8f09edf49d
2 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
MANIFEST.in
View file

@ -1,4 +1,5 @@
include CMakeLists.txt
include LICENSE
recursive-include src *
recursive-include python *
recursive-include single_include *

15
python/test/test_kompute.py
View file

@ -1,9 +1,12 @@
import os
from pyshader import python2shader, f32, ivec3, Array
from pyshader.stdlib import exp, log
from kp import Tensor, Manager, Sequence
DIRNAME = os.path.dirname(os.path.abspath(__file__))
def test_opmult():
"""
Test basic OpMult operation
@ -70,7 +73,7 @@ def test_opalgobase_file():
mgr = Manager()
shaderFilePath = "../../shaders/glsl/opmult.comp"
shaderFilePath = os.path.join(DIRNAME, "../../shaders/glsl/opmult.comp")
mgr.eval_tensor_create_def([tensor_in_a, tensor_in_b, tensor_out])
@ -85,20 +88,26 @@ def test_sequence():
Test basic OpAlgoBase operation
"""
mgr = Manager(0, [2])
tensor_in_a = Tensor([2, 2, 2])
tensor_in_b = Tensor([1, 2, 3])
tensor_out = Tensor([0, 0, 0])
mgr.eval_tensor_create_def([tensor_in_a, tensor_in_b, tensor_out])
seq = mgr.create_sequence("op")
shaderFilePath = "../../shaders/glsl/opmult.comp"
shaderFilePath = os.path.join(DIRNAME, "../../shaders/glsl/opmult.comp")
mgr.eval_async_algo_file_def([tensor_in_a, tensor_in_b, tensor_out], shaderFilePath)
mgr.eval_await_def()
seq = mgr.create_sequence("op")
seq.begin()
seq.record_tensor_sync_local([tensor_in_a])
seq.record_tensor_sync_local([tensor_in_b])
seq.record_tensor_sync_local([tensor_out])
seq.end()
seq.eval()
assert tensor_out.data() == [2.0, 4.0, 6.0]
def test_pyshader_pyshader():