mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-09-10 17:53:42 +08:00
github-actions[bot]
0e51ecbb7e
version: sunnypilot v2026.003.000 (dev)
date: 2026-09-09T15:43:49
master commit: b255314f5a
14 lines
722 B
Python
14 lines
722 B
Python
import numpy as np
|
|
from tinygrad.helpers import flat_mv
|
|
from tinygrad.device import Compiled, Allocator
|
|
|
|
class NpyAllocator(Allocator['NpyDevice']):
|
|
def _alloc(self, size:int, options=None) -> np.ndarray: return np.empty(size, dtype=np.uint8)
|
|
def _as_buffer(self, src:np.ndarray) -> memoryview: return flat_mv(np.require(src, requirements='C').data)
|
|
def _copyout(self, dest:memoryview, src:np.ndarray): dest[:] = self._as_buffer(src)
|
|
def _offset(self, buf:np.ndarray, size:int, offset:int) -> np.ndarray:
|
|
return np.require(buf, requirements='C').reshape(-1).view(np.uint8)[offset:offset+size]
|
|
|
|
class NpyDevice(Compiled):
|
|
def __init__(self, device:str): super().__init__(device, NpyAllocator(self), [], None)
|