def test_alloc():
c = get_core()
(r21,s21,w01,w2x) = c.acquire_fpregisters(range(4))
(i0,i1,ir0,is0,sixteen) = map(IntRegister,range(5))
c.name_registers(a21=FPRegister(4),b21=FPRegister(5))
istream = [
# weights
isa.fpset2(w01,1/9,2/9),
isa.fpset2(w2x,1/9,-1),
# preamble
isa.lfpd('a21',i0,0), # A[0],A[1]
isa.lfdu('a21',i0,16), # A[2],A[1]
isa.lfpd('b21',i1,0), # A[0],A[1]
isa.lfdu('b21',i1,16), # A[2],A[1]
# start loads
isa.fxcpmadd(r21,w01,'a21',r21),
isa.fxcpmadd(s21,w01,'b21',s21),
isa.lfpd('a23',i0,16),
isa.lfpd('b23',i1,16),
isa.fxcxma(r21,w01,'a23',r21),
isa.fxcxma(s21,w01,'b23',s21),
isa.lfdu('a23',i0,16), # a43
isa.lfdu('b23',i1,16), # b43
isa.intset(ir0,8*8),
isa.intset(is0,18*8),
isa.intset(sixteen,16),
isa.stfxdux(r21,ir0,sixteen),
isa.stfxdux(s21,is0,sixteen),
]
#for instr in istream: print(instr); #c.trace = c.trace_none
c.schedule(istream)
c.execute([isa.inspect()])
def test1():
c = get_core()
(r21,s21,w01,w2x,a21,b21,a23,b23) = c.acquire_fpregisters(range(8))
(i0,i1,ir0,is0,sixteen) = map(IntRegister,range(5))
istream = (s_weights(w01,w2x)
+ merge(s_preamble(a21,i0),s_preamble(b21,i1))
+ [
isa.fxcpmadd(r21,w01,a21,r21),
isa.fxcpmadd(s21,w01,b21,s21),
isa.lfpd(a23,i0,16),
isa.lfpd(b23,i1,16),
isa.fxcxma(r21,w01,a23,r21),
isa.fxcxma(s21,w01,b23,s21),
isa.lfdu(a23,i0,16), # Rename to a43
isa.lfdu(b23,i1,16), # Rename to a43
isa.intset(ir0,8*8),
isa.intset(is0,18*8),
isa.intset(sixteen,16),
isa.stfxdux(r21,ir0,sixteen),
isa.stfxdux(s21,is0,sixteen),
])
#for instr in istream: print(instr); #c.trace = c.trace_none
c.schedule(istream)
c.execute([isa.inspect()])
