mirror of
https://gitlvb.teallvbs.xyz/IQ.Lvbs/IQ.Pilot.git
synced 2026-07-24 13:02:07 +08:00
IQ.Lvbs CI [bot]
24 lines
779 B
Python
24 lines
779 B
Python
"""
|
|
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
|
|
"""
|
|
from tinygrad.tensor import Tensor
|
|
from tinygrad.helpers import to_mv
|
|
|
|
_PTR_STRIDE = 8
|
|
_RAW_GPU_PTR_SLOT = 20
|
|
_RAW_GPU_PTR_VIEW_BYTES = 0x100
|
|
|
|
|
|
def _descriptor_pointer(opencl_address: int) -> int:
|
|
return to_mv(opencl_address, _PTR_STRIDE).cast("Q")[0]
|
|
|
|
|
|
def _raw_gpu_pointer(descriptor_pointer: int) -> int:
|
|
return to_mv(descriptor_pointer, _RAW_GPU_PTR_VIEW_BYTES).cast("Q")[_RAW_GPU_PTR_SLOT]
|
|
|
|
|
|
def qcom_tensor_from_opencl_address(opencl_address, shape, dtype):
|
|
descriptor_pointer = _descriptor_pointer(opencl_address)
|
|
device_pointer = _raw_gpu_pointer(descriptor_pointer)
|
|
return Tensor.from_blob(device_pointer, shape, dtype=dtype, device="QCOM")
|