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 CMakeLists.txt
include LICENSE
recursive-include src * recursive-include src *
recursive-include python * recursive-include python *
recursive-include single_include * 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 import python2shader, f32, ivec3, Array
from pyshader.stdlib import exp, log from pyshader.stdlib import exp, log
from kp import Tensor, Manager, Sequence from kp import Tensor, Manager, Sequence
DIRNAME = os.path.dirname(os.path.abspath(__file__))
def test_opmult(): def test_opmult():
""" """
Test basic OpMult operation Test basic OpMult operation
@ -70,7 +73,7 @@ def test_opalgobase_file():
mgr = Manager() 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]) mgr.eval_tensor_create_def([tensor_in_a, tensor_in_b, tensor_out])
@ -85,20 +88,26 @@ def test_sequence():
Test basic OpAlgoBase operation Test basic OpAlgoBase operation
""" """
mgr = Manager(0, [2]) mgr = Manager(0, [2])
tensor_in_a = Tensor([2, 2, 2]) tensor_in_a = Tensor([2, 2, 2])
tensor_in_b = Tensor([1, 2, 3]) tensor_in_b = Tensor([1, 2, 3])
tensor_out = Tensor([0, 0, 0]) tensor_out = Tensor([0, 0, 0])
mgr.eval_tensor_create_def([tensor_in_a, tensor_in_b, tensor_out]) 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_async_algo_file_def([tensor_in_a, tensor_in_b, tensor_out], shaderFilePath)
mgr.eval_await_def() mgr.eval_await_def()
seq = mgr.create_sequence("op")
seq.begin() seq.begin()
seq.record_tensor_sync_local([tensor_in_a]) seq.record_tensor_sync_local([tensor_in_a])
seq.record_tensor_sync_local([tensor_in_b]) seq.record_tensor_sync_local([tensor_in_b])
seq.record_tensor_sync_local([tensor_out]) seq.record_tensor_sync_local([tensor_out])
seq.end() seq.end()
seq.eval() seq.eval()
assert tensor_out.data() == [2.0, 4.0, 6.0] assert tensor_out.data() == [2.0, 4.0, 6.0]
def test_pyshader_pyshader(): def test_pyshader_pyshader():