mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-11 18:53:47 +08:00
github-actions[bot]
eefc084302
version: sunnypilot v2025.003.000 (dev) date: 2025-12-18T05:48:43 master commit: 16c052af0835f049a67b80251d8cc1114fc08bb4
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')
|