mirror of
https://github.com/MoreTore/openpilot.git
synced 2026-08-22 08:33:44 +08:00
Harald Schäfer
7b5a4fbb03
* squash * bump tg * bump tg * debump tinygrad * bump tinygrad * bump tg * Skip init iteration * fixes * cleanups * skip first test sample * typos * linter unhappy * update cpu usage * OPENCL just zeros for now * imports * Try printing * Runs again, but slower * unused import * Allow more buffer with tg and all on gpu * bump tinygrad * seems ok * stricter timings for driving looser for dm * try llvm * check nvidia * More timeout for now * make test pass * Revert "try llvm" This reverts commit ef136e478320101fea262bae3579e558da991902. * small fixes * whitespace * revert test timeout * No model runners * Always CPU always fast * No onnx runtime GPU * more cores * cleanup * Is this faster * Is this faster * at least runs * FP32 is faster than 16 * fix deps * whitespace * comment --------- Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com>
9 lines
365 B
Python
9 lines
365 B
Python
|
|
from tinygrad.tensor import Tensor
|
|
from tinygrad.helpers import to_mv
|
|
|
|
def qcom_tensor_from_opencl_address(opencl_address, shape, dtype):
|
|
cl_buf_desc_ptr = to_mv(opencl_address, 8).cast('Q')[0]
|
|
rawbuf_ptr = to_mv(cl_buf_desc_ptr, 0x100).cast('Q')[20] # offset 0xA0 is a raw gpu pointer.
|
|
return Tensor.from_blob(rawbuf_ptr, shape, dtype=dtype, device='QCOM')
|