baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add validation for tensor_split size exceeding LLAMA_MAX_DEVICES (#820)

Browse source 
* Add validation for tensor_split size exceeding LLAMA_MAX_DEVICES

* reword
This commit is contained in:
Eric Liu 2023-10-15 10:51:51 -07:00 committed by GitHub
parent f30aa20126
commit b50166500e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
llama_cpp/llama.py
View file

@ -308,6 +308,8 @@ class Llama:
self.tensor_split = tensor_split self.tensor_split = tensor_split
self._p_tensor_split = None self._p_tensor_split = None
if self.tensor_split is not None: if self.tensor_split is not None:
if len(self.tensor_split) > llama_cpp.LLAMA_MAX_DEVICES:
raise ValueError(f"Attempt to split tensors that exceed maximum supported devices. Current LLAMA_MAX_DEVICES={llama_cpp.LLAMA_MAX_DEVICES}")
# Type conversion and expand the list to the length of LLAMA_MAX_DEVICES # Type conversion and expand the list to the length of LLAMA_MAX_DEVICES
FloatArray = ctypes.c_float * llama_cpp.LLAMA_MAX_DEVICES FloatArray = ctypes.c_float * llama_cpp.LLAMA_MAX_DEVICES
self._c_tensor_split = FloatArray( self._c_tensor_split = FloatArray(