litedram/bench/arty.py

245 lines
8.9 KiB
Python
Executable File

#!/usr/bin/env python3
# This file is Copyright (c) 2020 Florent Kermarrec <florent@enjoy-digital.fr>
# License: BSD
import os
import argparse
from migen import *
from litex_boards.platforms import arty
from litex.soc.cores.clock import *
from litex.soc.interconnect.csr import *
from litex.soc.integration.soc_core import *
from litex.soc.integration.soc_sdram import *
from litex.soc.integration.builder import *
from litedram.phy import s7ddrphy
from litedram.modules import MT41K128M16
from liteeth.phy.mii import LiteEthPHYMII
# CRG ----------------------------------------------------------------------------------------------
class _CRG(Module, AutoCSR):
def __init__(self, platform, sys_clk_freq):
self.clock_domains.cd_sys_pll = ClockDomain()
self.clock_domains.cd_sys = ClockDomain()
self.clock_domains.cd_sys4x = ClockDomain(reset_less=True)
self.clock_domains.cd_sys4x_dqs = ClockDomain(reset_less=True)
self.clock_domains.cd_clk200 = ClockDomain()
self.clock_domains.cd_eth = ClockDomain()
# # #
self.submodules.main_pll = main_pll = S7PLL(speedgrade=-1)
self.comb += main_pll.reset.eq(~platform.request("cpu_reset"))
main_pll.register_clkin(platform.request("clk100"), 100e6)
main_pll.create_clkout(self.cd_sys_pll, 100e6)
main_pll.create_clkout(self.cd_clk200, 200e6)
main_pll.create_clkout(self.cd_eth, 25e6)
main_pll.expose_drp()
self.submodules.idelayctrl = S7IDELAYCTRL(self.cd_clk200)
self.comb += platform.request("eth_ref_clk").eq(self.cd_eth.clk)
sys_clk_counter = Signal(32)
self.sync += sys_clk_counter.eq(sys_clk_counter + 1)
self.sys_clk_counter = CSRStatus(32)
self.comb += self.sys_clk_counter.status.eq(sys_clk_counter)
self.submodules.pll = pll = S7PLL(speedgrade=-1)
self.comb += pll.reset.eq(~main_pll.locked)
pll.register_clkin(self.cd_sys_pll.clk, 100e6)
pll.create_clkout(self.cd_sys, sys_clk_freq)
pll.create_clkout(self.cd_sys4x, 4*sys_clk_freq)
pll.create_clkout(self.cd_sys4x_dqs, 4*sys_clk_freq, phase=90)
# Bench SoC ----------------------------------------------------------------------------------------
class BenchSoC(SoCCore):
def __init__(self, sys_clk_freq=int(100e6)):
platform = arty.Platform()
# SoCCore ----------------------------------------------------------------------------------
SoCCore.__init__(self, platform, clk_freq=sys_clk_freq,
integrated_rom_size = 0x8000,
csr_data_width = 32,
uart_name = "crossover")
# CRG --------------------------------------------------------------------------------------
self.submodules.crg = _CRG(platform, sys_clk_freq)
self.add_csr("crg")
# DDR3 SDRAM -------------------------------------------------------------------------------
self.submodules.ddrphy = s7ddrphy.A7DDRPHY(platform.request("ddram"),
memtype = "DDR3",
nphases = 4,
sys_clk_freq = sys_clk_freq)
self.add_csr("ddrphy")
self.add_sdram("sdram",
phy = self.ddrphy,
module = MT41K128M16(sys_clk_freq, "1:4"),
origin = self.mem_map["main_ram"]
)
# Etherbone --------------------------------------------------------------------------------
self.submodules.ethphy = LiteEthPHYMII(
clock_pads = self.platform.request("eth_clocks"),
pads = self.platform.request("eth"),
with_hw_init_reset = False)
self.add_csr("ethphy")
self.add_etherbone(phy=self.ethphy)
# Leds -------------------------------------------------------------------------------------
from litex.soc.cores.led import LedChaser
self.submodules.led = LedChaser(self.platform.request_all("user_led"), sys_clk_freq)
self.add_csr("led")
# Bench Test ---------------------------------------------------------------------------------------
def bench_test():
import time
from litex import RemoteClient
wb = RemoteClient()
wb.open()
# # #
class ClkReg1:
def __init__(self, value=0):
self.unpack(value)
def unpack(self, value):
self.low_time = (value >> 0) & (2**6 - 1)
self.high_time = (value >> 6) & (2**6 - 1)
self.reserved = (value >> 12) & (2**1 - 1)
self.phase_mux = (value >> 13) & (2**3 - 1)
def pack(self):
value = (self.low_time << 0)
value |= (self.high_time << 6)
value |= (self.reserved << 12)
value |= (self.phase_mux << 13)
return value
def __repr__(self):
s = "ClkReg1:\n"
s += " low_time: {:d}\n".format(self.low_time)
s += " high_time: {:d}\n".format(self.high_time)
s += " reserved: {:d}\n".format(self.reserved)
s += " phase_mux: {:d}".format(self.phase_mux)
return s
class ClkReg2:
def __init__(self, value = 0):
self.unpack(value)
def unpack(self, value):
self.delay_time = (value >> 0) & (2**6 - 1)
self.no_count = (value >> 6) & (2**1 - 1)
self.edge = (value >> 7) & (2**1 - 1)
self.mx = (value >> 8) & (2**2 - 1)
self.frac_wf_r = (value >> 10) & (2**1 - 1)
self.frac_en = (value >> 11) & (2**1 - 1)
self.frac = (value >> 12) & (2**3 - 1)
self.reserved = (value >> 15) & (2**1 - 1)
def pack(self):
value = (self.delay_time << 0)
value |= (self.no_count << 6)
value |= (self.edge << 7)
value |= (self.mx << 8)
value |= (self.frac_wf_r << 10)
value |= (self.frac_en << 11)
value |= (self.frac << 12)
value |= (self.reserved << 15)
return value
def __repr__(self):
s = "ClkReg2:\n"
s += " delay_time: {:d}\n".format(self.delay_time)
s += " no_count: {:d}\n".format(self.no_count)
s += " edge: {:d}\n".format(self.edge)
s += " mx: {:d}\n".format(self.mx)
s += " frac_wf_r: {:d}\n".format(self.frac_wf_r)
s += " frac_en: {:d}\n".format(self.frac_en)
s += " frac: {:d}\n".format(self.frac)
s += " reserved: {:d}".format(self.reserved)
return s
class S7PLL:
def reset(self):
wb.regs.crg_main_pll_drp_reset.write(1)
def read(self, adr):
wb.regs.crg_main_pll_drp_adr.write(adr)
wb.regs.crg_main_pll_drp_read.write(1)
return wb.regs.crg_main_pll_drp_dat_r.read()
def write(self, adr, value):
wb.regs.crg_main_pll_drp_adr.write(adr)
wb.regs.crg_main_pll_drp_dat_w.write(value)
wb.regs.crg_main_pll_drp_write.write(1)
# # #
vco_freq = 1.6e9
s7pll = S7PLL()
print("Dump Main PLL...")
clkout0_clkreg1 = ClkReg1(s7pll.read(0x08))
print(clkout0_clkreg1)
for i in range(12, 44, 2):
sys_clk_freq = vco_freq/i
print("Reconfig Main PLL to {}MHz...".format(sys_clk_freq/1e6))
clkout0_clkreg1.high_time = i//2 + i%2
clkout0_clkreg1.low_time = i//2
s7pll.write(0x08, clkout0_clkreg1.pack())
print("Measuring sys_clk...")
duration = 5e-1
start = wb.regs.crg_sys_clk_counter.read()
time.sleep(duration)
end = wb.regs.crg_sys_clk_counter.read()
print("sys_clk: {:3.2f}MHz".format((end-start)/(1e6*duration)))
print("Reset SoC and get BIOS log...")
wb.regs.ctrl_reset.write(1)
start = time.time()
while (time.time() - start) < 5:
if wb.regs.uart_xover_rxempty.read() == 0:
print("{:c}".format(wb.regs.uart_xover_rxtx.read()), end="")
# # #
wb.close()
# Main ---------------------------------------------------------------------------------------------
def main():
parser = argparse.ArgumentParser(description="LiteDRAM Bench on Arty A7")
parser.add_argument("--build", action="store_true", help="Build bitstream")
parser.add_argument("--load", action="store_true", help="Load bitstream")
parser.add_argument("--test", action="store_true", help="Run Test")
args = parser.parse_args()
if args.build or args.load:
soc = BenchSoC()
builder = Builder(soc, csr_csv="csr.csv")
builder.build(run=args.build)
if args.load:
prog = soc.platform.create_programmer()
prog.load_bitstream(os.path.join(builder.gateware_dir, soc.build_name + ".bit"))
if args.test:
bench_test()
if __name__ == "__main__":
main()