update pi_dcbz function
[soc.git] / src / soc / config / test / test_pi2ls.py
1 from nmigen import Signal, Module, Record
2 from nmigen.back.pysim import Simulator, Delay
3 from nmigen.compat.sim import run_simulation, Settle
4 from nmutil.formaltest import FHDLTestCase
5 from nmigen.cli import rtlil
6 import unittest
7 from soc.config.test.test_loadstore import TestMemPspec
8 from soc.config.loadstore import ConfigMemoryPortInterface
9
10
11 def wait_busy(port, no=False,debug=None):
12 cnt = 0
13 while True:
14 busy = yield port.busy_o
15 print("busy", no, busy, cnt, debug)
16 if bool(busy) == no:
17 break
18 yield
19 cnt += 1
20
21
22
23 def wait_addr(port,debug=None):
24 cnt = 0
25 while True:
26 addr_ok = yield port.addr_ok_o
27 print("addrok", addr_ok,cnt,debug)
28 if addr_ok:
29 break
30 yield
31 cnt += 1
32
33
34 def wait_ldok(port):
35 cnt = 0
36 while True:
37 ldok = yield port.ld.ok
38 exc_happened = yield port.exc_o.happened
39 print("ldok", ldok, "exception", exc_happened, "count", cnt)
40 cnt += 1
41 if ldok or exc_happened:
42 break
43 yield
44
45
46 def pi_st(port1, addr, data, datalen, msr_pr=0):
47
48 # have to wait until not busy
49 yield from wait_busy(port1, no=False) # wait until not busy
50
51 # set up a ST on the port. address first:
52 yield port1.is_st_i.eq(1) # indicate ST
53 yield port1.data_len.eq(datalen) # ST length (1/2/4/8)
54 yield port1.msr_pr.eq(msr_pr) # MSR PR bit (1==>virt, 0==>real)
55
56 yield port1.addr.data.eq(addr) # set address
57 yield port1.addr.ok.eq(1) # set ok
58 yield Settle()
59 yield from wait_addr(port1) # wait until addr ok
60 # yield # not needed, just for checking
61 # yield # not needed, just for checking
62 # assert "ST" for one cycle (required by the API)
63 yield port1.st.data.eq(data)
64 yield port1.st.ok.eq(1)
65 yield
66 yield port1.st.ok.eq(0)
67 yield from wait_busy(port1, True) # wait while busy
68
69 # copy of pi_st
70 def pi_dcbz(port1, addr, msr_pr=0):
71
72 # have to wait until not busy
73 yield from wait_busy(port1, no=False,debug="busy") # wait until not busy
74
75 # set up a ST on the port. address first:
76 yield port1.is_st_i.eq(1) # indicate ST
77 yield port1.msr_pr.eq(msr_pr) # MSR PR bit (1==>virt, 0==>real)
78
79 yield port1.is_dcbz.eq(1) # set dcbz
80
81 yield port1.addr.data.eq(addr) # set address
82 yield port1.addr.ok.eq(1) # set ok
83 yield Settle()
84
85 # guess: this is not needed
86 # yield from wait_addr(port1,debug="addr") # wait until addr ok
87
88 # just write some dummy data -- remove
89 print("dummy write begin")
90 yield port1.st.data.eq(0)
91 yield port1.st.ok.eq(1)
92 yield
93 yield port1.st.ok.eq(0)
94 print("dummy write end")
95
96 yield from wait_busy(port1, no=True, debug="not_busy") # wait while busy
97
98 # can go straight to reset.
99 yield port1.is_st_i.eq(0) # end
100 yield port1.addr.ok.eq(0) # set !ok
101 yield port1.is_dcbz.eq(0) # reset dcbz too
102
103
104 def pi_ld(port1, addr, datalen, msr_pr=0):
105
106 # have to wait until not busy
107 yield from wait_busy(port1, no=False) # wait until not busy
108
109 # set up a LD on the port. address first:
110 yield port1.is_ld_i.eq(1) # indicate LD
111 yield port1.data_len.eq(datalen) # LD length (1/2/4/8)
112 yield port1.msr_pr.eq(msr_pr) # MSR PR bit (1==>virt, 0==>real)
113
114 yield port1.addr.data.eq(addr) # set address
115 yield port1.addr.ok.eq(1) # set ok
116 yield Settle()
117 yield from wait_addr(port1) # wait until addr ok
118 yield
119 yield from wait_ldok(port1) # wait until ld ok
120 data = yield port1.ld.data
121 exc_happened = yield port1.exc_o.happened
122
123 # cleanup
124 yield port1.is_ld_i.eq(0) # end
125 yield port1.addr.ok.eq(0) # set !ok
126 if exc_happened:
127 return 0
128
129 yield from wait_busy(port1, no=False) # wait while not busy
130
131 return data
132
133
134 def pi_ldst(arg, dut, msr_pr=0):
135
136 # do two half-word stores at consecutive addresses, then two loads
137 addr1 = 0x04
138 addr2 = addr1 + 0x2
139 data = 0xbeef
140 data2 = 0xf00f
141 #data = 0x4
142 yield from pi_st(dut, addr1, data, 2, msr_pr)
143 yield from pi_st(dut, addr2, data2, 2, msr_pr)
144 result = yield from pi_ld(dut, addr1, 2, msr_pr)
145 result2 = yield from pi_ld(dut, addr2, 2, msr_pr)
146 arg.assertEqual(data, result, "data %x != %x" % (result, data))
147 arg.assertEqual(data2, result2, "data2 %x != %x" % (result2, data2))
148
149 # now load both in a 32-bit load to make sure they're really consecutive
150 data3 = data | (data2 << 16)
151 result3 = yield from pi_ld(dut, addr1, 4, msr_pr)
152 arg.assertEqual(data3, result3, "data3 %x != %x" % (result3, data3))
153
154
155 def tst_config_pi(testcls, ifacetype):
156 """set up a configureable memory test of type ifacetype
157 """
158 dut = Module()
159 pspec = TestMemPspec(ldst_ifacetype=ifacetype,
160 imem_ifacetype='',
161 addr_wid=48,
162 mask_wid=8,
163 reg_wid=64)
164 cmpi = ConfigMemoryPortInterface(pspec)
165 dut.submodules.pi = cmpi.pi
166 if hasattr(cmpi, 'lsmem'): # hmmm not happy about this
167 dut.submodules.lsmem = cmpi.lsmem.lsi
168 vl = rtlil.convert(dut, ports=[]) # dut.ports())
169 with open("test_pi_%s.il" % ifacetype, "w") as f:
170 f.write(vl)
171
172 run_simulation(dut, {"sync": pi_ldst(testcls, cmpi.pi.pi)},
173 vcd_name='test_pi_%s.vcd' % ifacetype)
174
175
176 class TestPIMem(unittest.TestCase):
177
178 def test_pi_mem(self):
179 tst_config_pi(self, 'testpi')
180
181 def test_pi2ls(self):
182 tst_config_pi(self, 'testmem')
183
184 def test_pi2ls_bare_wb(self):
185 tst_config_pi(self, 'test_bare_wb')
186
187
188 if __name__ == '__main__':
189 unittest.main()