1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
#!/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!")
|
