mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
firestar5683
22a8575199
Keep official openpilot ancestry and dependency licenses. Store resources directly in Git, remove LFS setup requirements, and leave the downloadable big model outside the source tree.
28 lines
988 B
Python
28 lines
988 B
Python
from tinygrad import Device
|
|
|
|
# PATH=/opt/homebrew/opt/llvm/bin:$PATH python3 extra/dsp/opt.py
|
|
|
|
if __name__ == "__main__":
|
|
compiler = Device["DSP"].compiler
|
|
|
|
lib = compiler.compile("""
|
|
typedef long HVX_Vector __attribute__((__vector_size__(128))) __attribute__ ((aligned(128)));
|
|
typedef long HVX_VectorPair __attribute__((__vector_size__(256))) __attribute__ ((aligned(256)));
|
|
|
|
void test(unsigned char *c, unsigned char *a, unsigned char *b) {
|
|
HVX_Vector t0 = *(HVX_Vector*)a;
|
|
//HVX_VectorPair t1 = *((HVX_VectorPair*)b);
|
|
HVX_Vector acc = __builtin_HEXAGON_V6_vd0_128B();
|
|
for (int i = 0; i < 128; i++) {
|
|
//__builtin_HEXAGON_V6_lvsplatb_128B(t0[i])
|
|
//acc += __builtin_HEXAGON_V6_lvsplatb_128B(t0[i]) * t1;
|
|
//acc += t0[i] * t1;
|
|
unsigned int t1 = ((unsigned int *)b)[i];
|
|
//acc = __builtin_HEXAGON_V6_vrmpyub_acc_128B(acc, t0, t1);
|
|
acc = __builtin_HEXAGON_V6_vrmpybus_acc_128B(acc, t0, t1);
|
|
}
|
|
*((HVX_Vector*)c) = acc;
|
|
}""")
|
|
|
|
compiler.disassemble(lib)
|