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

Added list_devices function and separated new utils function

Browse source
This commit is contained in:
Alejandro Saucedo 2021-03-20 08:49:51 +00:00
parent d23ea95e64
commit a6457ee1ec
3 changed files with 49 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

12
python/test/test_kompute.py
View file

@ -227,3 +227,15 @@ def test_workgroup():
assert np.all(tensor_a.data() == np.stack([np.arange(16)]*8, axis=1).ravel())
assert np.all(tensor_b.data() == np.stack([np.arange(8)]*16, axis=0).ravel())
def test_mgr_utils():
mgr = kp.Manager()
props = mgr.get_device_properties()
assert "device_name" in props
devices = mgr.list_devices()
assert len(devices) == 1
assert "device_name" in devices[0]