mirror of
https://github.com/AsahiLinux/m1n1
synced 2024-11-30 02:10:16 +00:00
44 lines
1 KiB
Python
44 lines
1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# SPDX-License-Identifier: MIT
|
||
|
|
import sys, pathlib
|
||
|
|
sys.path.append(str(pathlib.Path(__file__).resolve().parents[1]))
|
||
|
|
|
||
|
|
from m1n1.setup import *
|
||
|
|
from m1n1.shell import run_shell
|
||
|
|
|
||
|
|
from m1n1.fw.ane import ANE
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
from anect import anect_convert # pip install anect
|
||
|
|
def f16encode(x): return np.float16(x).tobytes()
|
||
|
|
def f16decode(b): return np.frombuffer(b[:2], dtype=np.float16)[0]
|
||
|
|
|
||
|
|
|
||
|
|
ane = ANE(u)
|
||
|
|
ane.power_up()
|
||
|
|
if 1:
|
||
|
|
rnges = [(0x26bc04000, 0x26bc28000, 'engine'),]
|
||
|
|
mon = RegMonitor(u)
|
||
|
|
for (start, end, name) in rnges:
|
||
|
|
mon.add(start, end-start, name=name)
|
||
|
|
mon.poll() # should work after ane.power_up()
|
||
|
|
|
||
|
|
# curl -LJO https://www.dropbox.com/s/lpjap6w0kdlom1h/add.hwx?dl=0
|
||
|
|
anec = anect_convert("add.hwx")
|
||
|
|
req = ane.fw.setup(anec)
|
||
|
|
|
||
|
|
x1 = f16encode(1.0)
|
||
|
|
x2 = f16encode(2.0)
|
||
|
|
ane.fw.send_src(req, x1, 0)
|
||
|
|
ane.fw.send_src(req, x2, 1)
|
||
|
|
|
||
|
|
ane.tm.enqueue_tq(req)
|
||
|
|
ane.tm.execute_tq(req)
|
||
|
|
|
||
|
|
x3 = ane.fw.read_dst(req, 0)
|
||
|
|
print("what's 1+2? = %f" % f16decode(x3[:2]))
|
||
|
|
|
||
|
|
|
||
|
|
run_shell(globals(), msg="Have fun!")
|
||
|
|
|