Refine input handling logic for non-TICI devices.

This change adds an explicit check for non-TICI devices in the input processing logic, ensuring correctness and preventing unnecessary operations. The updated logic improves maintainability and clarity in device-specific handling.
This commit is contained in:
DevTekVE
2024-12-30 15:16:38 +01:00
parent f72447245c
commit cfffc94a07
+1 -1
View File
@@ -71,7 +71,7 @@ class TinygradRunner(ModelRunner):
for key in imgs_cl:
if TICI and key not in self.inputs:
self.inputs[key] = qcom_tensor_from_opencl_address(imgs_cl[key].mem_address, self.input_shapes[key], dtype=self.input_to_dtype[key])
else:
elif not TICI:
shape = self.frames[key].buffer_from_cl(imgs_cl[key]).reshape(self.input_shapes[key])
self.inputs[key] = Tensor(shape, device=self.input_to_device[key], dtype=self.input_to_dtype[key]).realize()