Files
tinygrad/accel/rawcpu/cbuffer.pxd
George Hotz 783c120a8c rawcpu (#365)
* rawcpu

* add should work when we respect shapetracker

* now that's true

* still have to handle shapetracker

* copyin

* Fix mypy
2022-08-17 11:33:20 +02:00

13 lines
269 B
Cython

# distutils: language = c++
# distutils: sources = cbuffer.h
cdef extern from "cbuffer.h":
cdef cppclass CBuffer:
CBuffer(int size)
void copyin(void *dat)
void add(CBuffer *a, CBuffer *b)
void mul(CBuffer *a, CBuffer *b)
float *buf
int size