mirror of
https://github.com/openhwgroup/cvw.git
synced 2025-04-25 06:17:10 -04:00
Refactor wsim to use smaller functions and f-strings
This commit is contained in:
Jordan Carlin
parent
cd90e81c76
commit
fbe3254857
1 changed files with 134 additions and 130 deletions
264
bin/wsim
264
bin/wsim
|
|
@ -14,160 +14,164 @@
|
||||||
import argparse
|
import argparse
|
||||||
import os
|
import os
|
||||||
|
|
||||||
########################
|
# Global variable
|
||||||
# main wsim script
|
|
||||||
########################
|
|
||||||
|
|
||||||
# Parse arguments
|
|
||||||
parser = argparse.ArgumentParser()
|
|
||||||
parser.add_argument("config", help="Configuration file")
|
|
||||||
parser.add_argument("testsuite", help="Test suite or path to .elf file")
|
|
||||||
parser.add_argument("--elf", "-e", help="ELF File name; use if name does not end in .elf", default="")
|
|
||||||
parser.add_argument("--sim", "-s", help="Simulator", choices=["questa", "verilator", "vcs"], default="questa")
|
|
||||||
parser.add_argument("--tb", "-t", help="Testbench", choices=["testbench", "testbench_fp"], default="testbench")
|
|
||||||
parser.add_argument("--gui", "-g", help="Simulate with GUI", action="store_true")
|
|
||||||
parser.add_argument("--ccov", "-c", help="Code Coverage", action="store_true")
|
|
||||||
parser.add_argument("--fcov", "-f", help="Functional Coverage with cvw-arch-verif, implies lockstep", action="store_true")
|
|
||||||
parser.add_argument("--args", "-a", help="Optional arguments passed to simulator via $value$plusargs", default="")
|
|
||||||
parser.add_argument("--params", "-p", help="Optional top-level parameter overrides of the form param=value", default="")
|
|
||||||
parser.add_argument("--vcd", "-v", help="Generate testbench.vcd", action="store_true")
|
|
||||||
parser.add_argument("--lockstep", "-l", help="Run ImperasDV lock, step, and compare.", action="store_true")
|
|
||||||
parser.add_argument("--locksteplog", "-b", help="Retired instruction number to be begin logging.", default=0)
|
|
||||||
parser.add_argument("--lockstepverbose", "-lv", help="Run ImperasDV lock, step, and compare with tracing enabled", action="store_true")
|
|
||||||
parser.add_argument("--covlog", "-d", help="Log coverage after n instructions.", default=0)
|
|
||||||
parser.add_argument("--rvvi", "-r", help="Simulate rvvi hardware interface and ethernet.", action="store_true")
|
|
||||||
args = parser.parse_args()
|
|
||||||
print("Config=" + args.config + " tests=" + args.testsuite + " sim=" + args.sim + " gui=" + str(args.gui) + " args='" + args.args + "'")
|
|
||||||
ElfFile=""
|
|
||||||
WALLY = os.environ.get('WALLY')
|
WALLY = os.environ.get('WALLY')
|
||||||
|
|
||||||
if(os.path.isfile(args.elf)):
|
def parseArgs():
|
||||||
ElfFile = "+ElfFile=" + os.path.abspath(args.elf)
|
parser = argparse.ArgumentParser()
|
||||||
elif (args.elf != ""):
|
parser.add_argument("config", help="Configuration file")
|
||||||
print("ELF file not found: " + args.elf)
|
parser.add_argument("testsuite", help="Test suite or path to .elf file")
|
||||||
exit(1)
|
parser.add_argument("--elf", "-e", help="ELF File name; use if name does not end in .elf", default="")
|
||||||
|
parser.add_argument("--sim", "-s", help="Simulator", choices=["questa", "verilator", "vcs"], default="questa")
|
||||||
|
parser.add_argument("--tb", "-t", help="Testbench", choices=["testbench", "testbench_fp"], default="testbench")
|
||||||
|
parser.add_argument("--gui", "-g", help="Simulate with GUI", action="store_true")
|
||||||
|
parser.add_argument("--ccov", "-c", help="Code Coverage", action="store_true")
|
||||||
|
parser.add_argument("--fcov", "-f", help="Functional Coverage with cvw-arch-verif, implies lockstep", action="store_true")
|
||||||
|
parser.add_argument("--args", "-a", help="Optional arguments passed to simulator via $value$plusargs", default="")
|
||||||
|
parser.add_argument("--params", "-p", help="Optional top-level parameter overrides of the form param=value", default="")
|
||||||
|
parser.add_argument("--vcd", "-v", help="Generate testbench.vcd", action="store_true")
|
||||||
|
parser.add_argument("--lockstep", "-l", help="Run ImperasDV lock, step, and compare.", action="store_true")
|
||||||
|
parser.add_argument("--locksteplog", "-b", help="Retired instruction number to be begin logging.", default=0)
|
||||||
|
parser.add_argument("--lockstepverbose", "-lv", help="Run ImperasDV lock, step, and compare with tracing enabled", action="store_true")
|
||||||
|
parser.add_argument("--covlog", "-d", help="Log coverage after n instructions.", default=0)
|
||||||
|
parser.add_argument("--rvvi", "-r", help="Simulate rvvi hardware interface and ethernet.", action="store_true")
|
||||||
|
return parser.parse_args()
|
||||||
|
|
||||||
if(args.testsuite.endswith('.elf') and args.elf == ""): # No --elf argument; check if testsuite has a .elf extension and use that instead
|
def elfFileCheck(args):
|
||||||
if (os.path.isfile(args.testsuite)):
|
ElfFile = ""
|
||||||
ElfFile = "+ElfFile=" + os.path.abspath(args.testsuite)
|
if os.path.isfile(args.elf):
|
||||||
# extract the elf name from the path to be the test suite
|
ElfFile = f"+ElfFile={os.path.abspath(args.elf)}"
|
||||||
fields = args.testsuite.rsplit('/', 3)
|
elif args.elf != "":
|
||||||
# if the name is just ref.elf in a deep path (riscv-arch-test/wally-riscv-arch-test), then use the directory name as the test suite to make it unique; otherwise work directory will have duplicates.
|
print(f"ELF file not found: {args.elf}")
|
||||||
if (len(fields) > 3):
|
exit(1)
|
||||||
if (fields[2] == "ref"):
|
elif args.testsuite.endswith('.elf'): # No --elf argument; check if testsuite has a .elf extension and use that instead
|
||||||
args.testsuite = fields[1] + "_" + fields[3]
|
if os.path.isfile(args.testsuite):
|
||||||
else:
|
ElfFile = f"+ElfFile={os.path.abspath(args.testsuite)}"
|
||||||
args.testsuite = fields[2] + "_" + fields[3]
|
# extract the elf name from the path to be the test suite
|
||||||
elif ('/' in args.testsuite):
|
fields = args.testsuite.rsplit('/', 3)
|
||||||
args.testsuite=args.testsuite.rsplit('/', 1)[1] # strip off path if present
|
# if the name is just ref.elf in a deep path (riscv-arch-test/wally-riscv-arch-test), then use the directory name as the test suite to make it unique; otherwise work directory will have duplicates.
|
||||||
else:
|
if (len(fields) > 3):
|
||||||
print("ELF file not found: " + args.testsuite)
|
if (fields[2] == "ref"):
|
||||||
|
args.testsuite = f"{fields[1]}_{fields[3]}"
|
||||||
|
else:
|
||||||
|
args.testsuite = f"{fields[2]}_{fields[3]}"
|
||||||
|
elif ('/' in args.testsuite):
|
||||||
|
args.testsuite=args.testsuite.rsplit('/', 1)[1] # strip off path if present
|
||||||
|
else:
|
||||||
|
print(f"ELF file not found: {args.testsuite}")
|
||||||
|
exit(1)
|
||||||
|
return ElfFile
|
||||||
|
|
||||||
|
def validateArgs(args):
|
||||||
|
if(args.lockstep and not args.testsuite.endswith('.elf') and not args.testsuite == "buildroot"):
|
||||||
|
print(f"Invalid Options. Cannot run a testsuite, {args.testsuite} with lockstep. Must run a single elf.")
|
||||||
|
exit(1)
|
||||||
|
elif (args.gui or args.ccov or args.fcov or args.lockstep or args.lockstepverbose) and args.sim not in ["questa", "vcs"]:
|
||||||
|
print("Option only supported for Questa and VCS")
|
||||||
|
exit(1)
|
||||||
|
elif (args.tb == "testbench_fp" and args.sim != "questa"):
|
||||||
|
print("Error: testbench_fp presently only supported by Questa, not VCS or Verilator, because of a touchy testbench")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
||||||
if (ElfFile != ""):
|
def prepSim(args, ElfFile):
|
||||||
args.args += " " + ElfFile
|
flags = ""
|
||||||
|
if args.vcd:
|
||||||
|
args.args += " -DMAKEVCD=1"
|
||||||
|
if args.rvvi:
|
||||||
|
args.params += " RVVI_SYNTH_SUPPORTED=1 "
|
||||||
|
if args.tb == "testbench_fp":
|
||||||
|
args.params += f" TEST=\" {args.testsuite} \" "
|
||||||
|
if ElfFile != "":
|
||||||
|
args.args += f" {ElfFile}"
|
||||||
|
if args.ccov:
|
||||||
|
flags += " --ccov"
|
||||||
|
if args.fcov:
|
||||||
|
flags += " --fcov"
|
||||||
|
prefix, suffix = lockstepSetup(args)
|
||||||
|
flags += suffix
|
||||||
|
return flags, prefix
|
||||||
|
|
||||||
if(args.lockstep and not args.testsuite.endswith('.elf') and not args.testsuite == "buildroot"):
|
def lockstepSetup(args):
|
||||||
print(f"Invalid Options. Cannot run a testsuite, {args.testsuite} with lockstep. Must run a single elf.")
|
prefix = ""
|
||||||
exit(1)
|
|
||||||
|
|
||||||
# Validate arguments
|
|
||||||
if (args.gui or args.ccov or args.fcov or args.lockstep or args.lockstepverbose) and args.sim not in ["questa", "vcs"]:
|
|
||||||
print("Option only supported for Questa and VCS")
|
|
||||||
exit(1)
|
|
||||||
elif (args.tb == "testbench_fp" and args.sim != "questa"):
|
|
||||||
print("Error: testbench_fp presently only supported by Questa, not VCS or Verilator, because of a touchy testbench")
|
|
||||||
exit(1)
|
|
||||||
|
|
||||||
if (args.vcd):
|
|
||||||
args.args += " -DMAKEVCD=1"
|
|
||||||
|
|
||||||
if (args.rvvi):
|
|
||||||
args.params += " RVVI_SYNTH_SUPPORTED=1 "
|
|
||||||
|
|
||||||
if (args.tb == "testbench_fp"):
|
|
||||||
args.params += " TEST=\"" + args.testsuite + "\" "
|
|
||||||
|
|
||||||
# if lockstep is enabled, then we need to pass the Imperas lockstep arguments
|
|
||||||
if(int(args.locksteplog) >= 1): EnableLog = 1
|
|
||||||
else: EnableLog = 0
|
|
||||||
prefix = ""
|
|
||||||
if (args.lockstep or args.lockstepverbose or args.fcov):
|
|
||||||
imperasicPath = os.path.join(WALLY, "config", args.config, "imperas.ic")
|
|
||||||
if not os.path.isfile(imperasicPath): # If config is a derivative, look for imperas.ic in derivative configs
|
|
||||||
imperasicPath = os.path.join(WALLY, "config", "deriv", args.config, "imperas.ic")
|
|
||||||
prefix = "IMPERAS_TOOLS=" + imperasicPath
|
|
||||||
|
|
||||||
if (args.lockstep or args.lockstepverbose):
|
|
||||||
if(args.locksteplog != 0): ImperasPlusArgs = " +IDV_TRACE2LOG=" + str(EnableLog) + " +IDV_TRACE2LOG_AFTER=" + str(args.locksteplog)
|
|
||||||
else: ImperasPlusArgs = ""
|
|
||||||
if(args.fcov):
|
|
||||||
CovEnableStr = "1" if int(args.covlog) > 0 else "0"
|
|
||||||
if(args.covlog >= 1): EnableLog = 1
|
|
||||||
else: EnableLog = 0
|
|
||||||
ImperasPlusArgs = " +IDV_TRACE2COV=" + str(EnableLog) + " +TRACE2LOG_AFTER=" + str(args.covlog) + " +TRACE2COV_ENABLE=" + CovEnableStr
|
|
||||||
suffix = ""
|
|
||||||
else:
|
|
||||||
CovEnableStr = ""
|
|
||||||
suffix = "--lockstep"
|
|
||||||
if(args.lockstepverbose):
|
|
||||||
prefix += ":" + WALLY + "/sim/imperas-verbose.ic"
|
|
||||||
else:
|
|
||||||
ImperasPlusArgs = ""
|
|
||||||
suffix = ""
|
suffix = ""
|
||||||
flags = suffix
|
ImperasPlusArgs = ""
|
||||||
args.args += ImperasPlusArgs
|
|
||||||
|
|
||||||
|
if(int(args.locksteplog) >= 1): EnableLog = 1
|
||||||
|
else: EnableLog = 0
|
||||||
|
if (args.lockstep or args.lockstepverbose or args.fcov):
|
||||||
|
imperasicPath = os.path.join(WALLY, "config", args.config, "imperas.ic")
|
||||||
|
if not os.path.isfile(imperasicPath): # If config is a derivative, look for imperas.ic in derivative configs
|
||||||
|
imperasicPath = os.path.join(WALLY, "config", "deriv", args.config, "imperas.ic")
|
||||||
|
if not os.path.isfile(imperasicPath):
|
||||||
|
print("Error: imperas.ic not found")
|
||||||
|
exit(1)
|
||||||
|
prefix += f"IMPERAS_TOOLS= {imperasicPath}"
|
||||||
|
|
||||||
# other flags
|
if (args.lockstep or args.lockstepverbose):
|
||||||
if (args.ccov):
|
if(args.locksteplog != 0): ImperasPlusArgs = f" +IDV_TRACE2LOG={EnableLog} +IDV_TRACE2LOG_AFTER={args.locksteplog}"
|
||||||
flags += " --ccov"
|
if(args.fcov):
|
||||||
if (args.fcov):
|
CovEnableStr = "1" if int(args.covlog) > 0 else "0"
|
||||||
flags += " --fcov"
|
if(args.covlog >= 1): EnableLog = 1
|
||||||
|
else: EnableLog = 0
|
||||||
|
ImperasPlusArgs = f" +IDV_TRACE2COV={EnableLog} +TRACE2LOG_AFTER={args.covlog} +TRACE2COV_ENABLE={CovEnableStr}"
|
||||||
|
else:
|
||||||
|
suffix = "--lockstep"
|
||||||
|
if(args.lockstepverbose):
|
||||||
|
prefix += f":{WALLY}/sim/imperas-verbose.ic"
|
||||||
|
args.args += ImperasPlusArgs
|
||||||
|
return prefix, suffix
|
||||||
|
|
||||||
# create the output sub-directories.
|
def createDirs(args):
|
||||||
regressionDir = WALLY + '/sim/'
|
for d in ["logs", "wkdir", "cov", "ucdb", "fcov", "fcov_ucdb"]:
|
||||||
for d in ["logs", "wkdir", "cov", "ucdb", "fcov", "fcov_ucdb"]:
|
os.makedirs(os.path.join(WALLY, "sim", args.sim, d), exist_ok=True)
|
||||||
try:
|
|
||||||
os.mkdir(regressionDir+args.sim+"/"+d)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
cd = "cd $WALLY/sim/" +args.sim
|
def runSim(args, flags, prefix):
|
||||||
|
if (args.sim == "questa"):
|
||||||
|
runQuesta(args, flags, prefix)
|
||||||
|
elif (args.sim == "verilator"):
|
||||||
|
runVerilator(args, flags, prefix)
|
||||||
|
elif (args.sim == "vcs"):
|
||||||
|
runVCS(args, flags, prefix)
|
||||||
|
|
||||||
# per-simulator launch
|
def runQuesta(args, flags, prefix):
|
||||||
if (args.sim == "questa"):
|
|
||||||
# Force Questa to use 64-bit mode, sometimes it defaults to 32-bit even on 64-bit machines
|
# Force Questa to use 64-bit mode, sometimes it defaults to 32-bit even on 64-bit machines
|
||||||
prefix = "MTI_VCO_MODE=64 " + prefix
|
prefix = "MTI_VCO_MODE=64 " + prefix
|
||||||
if (args.gui) and (args.tb == "testbench"):
|
if (args.gui) and (args.tb == "testbench"):
|
||||||
args.params += "DEBUG=1"
|
args.params += "DEBUG=1"
|
||||||
if (args.args != ""):
|
if (args.args != ""):
|
||||||
args.args = " --args \\\"" + args.args + "\\\""
|
args.args = f" --args \\\"{args.args}\\\""
|
||||||
if (args.params != ""):
|
if (args.params != ""):
|
||||||
args.params = " --params \\\"" + args.params + "\\\""
|
args.params = f" --params \\\"{args.params}\\\""
|
||||||
# Questa cannot accept more than 9 arguments. fcov implies lockstep
|
# Questa cannot accept more than 9 arguments. fcov implies lockstep
|
||||||
cmd = "do wally.do " + args.config + " " + args.testsuite + " " + args.tb + " " + args.args + " " + args.params + " " + flags
|
cmd = f"do wally.do {args.config} {args.testsuite} {args.tb} {args.args} {args.params} {flags}"
|
||||||
if (args.gui): # launch Questa with GUI; add +acc to keep variables accessible
|
if (args.gui): # launch Questa with GUI; add +acc to keep variables accessible
|
||||||
cmd = cd + "; " + prefix + " vsim -do \"" + cmd + " +acc\""
|
cmd = f"cd $WALLY/sim/questa; {prefix} vsim -do \" {cmd} +acc\""
|
||||||
else: # launch Questa in batch mode
|
else: # launch Questa in batch mode
|
||||||
cmd = cd + "; " + prefix + " vsim -c -do \"" + cmd + "\""
|
cmd = f"cd $WALLY/sim/questa; {prefix} vsim -c -do \" {cmd} \""
|
||||||
print("Running Questa with command: " + cmd)
|
print(f"Running Questa with command: {cmd}")
|
||||||
os.system(cmd)
|
os.system(cmd)
|
||||||
elif (args.sim == "verilator"):
|
|
||||||
|
def runVerilator(args, flags, prefix):
|
||||||
print(f"Running Verilator on {args.config} {args.testsuite}")
|
print(f"Running Verilator on {args.config} {args.testsuite}")
|
||||||
os.system(f"/usr/bin/make -C {regressionDir}/verilator WALLYCONF={args.config} TEST={args.testsuite} TESTBENCH={args.tb} PLUS_ARGS=\"{args.args}\" PARAM_ARGS=\"{args.params}\"")
|
os.system(f"/usr/bin/make -C {WALLY}/sim/verilator WALLYCONF={args.config} TEST={args.testsuite} TESTBENCH={args.tb} PLUS_ARGS=\"{args.args}\" PARAM_ARGS=\"{args.params}\"")
|
||||||
elif (args.sim == "vcs"):
|
|
||||||
print(f"Running VCS on " + args.config + " " + args.testsuite)
|
def runVCS(args, flags, prefix):
|
||||||
|
print(f"Running VCS on {args.config} {args.testsuite}")
|
||||||
# if (args.gui):
|
# if (args.gui):
|
||||||
# flags += " --gui"
|
# flags += " --gui"
|
||||||
if (args.args == ""):
|
if (args.args != ""):
|
||||||
vcsargs = ""
|
args.args = f" --args \"{args.args}\" "
|
||||||
else:
|
if (args.params != ""):
|
||||||
vcsargs = " --args \"" + args.args + "\" "
|
args.params = f" --params \"{args.params}\" "
|
||||||
if (args.params == ""):
|
cmd = f"cd $WALLY/sim/vcs; {prefix} ./run_vcs {args.config} {args.testsuite} --tb {args.tb} {args.args} {args.params} {flags}"
|
||||||
vcsparams = ""
|
|
||||||
else:
|
|
||||||
vcsparams = " --params \"" + args.params + "\" "
|
|
||||||
cmd = cd + "; " + prefix + " ./run_vcs " + args.config + " " + args.testsuite + " " + " --tb " + args.tb + " " + vcsargs + vcsparams + " " + flags
|
|
||||||
print(cmd)
|
print(cmd)
|
||||||
os.system(cmd)
|
os.system(cmd)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
args = parseArgs()
|
||||||
|
print(f"Config={args.config} tests={args.testsuite} sim={args.sim} gui={args.gui} args='{args.args} params='{args.params}'")
|
||||||
|
ElfFile = elfFileCheck(args)
|
||||||
|
validateArgs(args)
|
||||||
|
flags, prefix = prepSim(args, ElfFile)
|
||||||
|
createDirs(args)
|
||||||
|
exit(runSim(args, flags, prefix))
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue