diff options
Diffstat (limited to '.ci/utils')
| -rw-r--r-- | .ci/utils/jtag/viv_hardware_utils.tcl | 97 | ||||
| -rw-r--r-- | .ci/utils/mutex_hardware.py | 78 |
2 files changed, 175 insertions, 0 deletions
diff --git a/.ci/utils/jtag/viv_hardware_utils.tcl b/.ci/utils/jtag/viv_hardware_utils.tcl new file mode 100644 index 000000000..49ad74c0f --- /dev/null +++ b/.ci/utils/jtag/viv_hardware_utils.tcl @@ -0,0 +1,97 @@ +# Function definitions +proc ::connect_server { {hostname localhost} {port 3121} } { + if { [string compare [current_hw_server -quiet] ""] != 0 } { + disconnect_server + } + connect_hw_server -url $hostname:$port +} + +proc ::disconnect_server { } { + disconnect_hw_server [current_hw_server] +} + +proc ::jtag_list {} { + # Iterate through all hardware targets + set hw_targets [get_hw_targets -of_objects [current_hw_server -quiet] -quiet] + set idx_t 0 + foreach hw_target $hw_targets { + puts "== Target${idx_t}: $hw_target ==" + open_hw_target $hw_target -quiet + # Iterate through all hardware devices + set hw_devices [get_hw_devices] + set idx_d 0 + foreach hw_device $hw_devices { + puts "--- Device${idx_d}: $hw_device (Address = ${idx_t}:${idx_d})" + set idx_d [expr $idx_d + 1] + } + close_hw_target -quiet + set idx_t [expr $idx_t + 1] + } +} + +proc ::jtag_program { filepath {serial "."} {address "0:0"} } { + set idx_t [lindex [split $address :] 0] + set idx_d [lindex [split $address :] 1] + + set hw_targets [get_hw_targets -of_objects [current_hw_server]] + set hw_targets_regexp {} + + foreach target $hw_targets { + if { [regexp $serial $target] } { + set hw_targets_regexp [concat $hw_targets_regexp $target] + } + } + + set hw_target [lindex $hw_targets_regexp $idx_t] + + if { [string compare $hw_target ""] == 0 } { + error "ERROR: Could not open hw_target $idx_t. Either the address $address is incorrect or the device is not connected." + } else { + open_hw_target $hw_target -quiet + } + + set hw_device [lindex [get_hw_devices] $idx_d] + if { [string compare $hw_device ""] == 0 } { + close_hw_target -quiet + error "ERROR: Could not open hw_device $idx_d. Either the address $address is incorrect or the device is not connected." + } else { + puts "- Target: $hw_target" + puts "- Device: $hw_device" + puts "- Filename: $filepath" + puts "Programming..." + current_hw_device $hw_device + set_property PROBES.FILE {} [current_hw_device] + set_property PROGRAM.FILE $filepath [current_hw_device] + program_hw_devices [current_hw_device] + close_hw_target -quiet + puts "Programming DONE" + } +} + +# Initialization sequence +open_hw_manager +connect_server + +if [expr $argc > 0] { + #Execute a command and exit + set cmd [lindex $argv 0] + if { [string compare $cmd "list"] == 0 } { + jtag_list + } elseif { [string compare $cmd "program"] == 0 } { + set filepath [lindex $argv 1] + if [expr $argc == 3] { + set serial [lindex $argv 2] + jtag_program $filepath $serial + } elseif [expr $argc > 3] { + set serial [lindex $argv 2] + set devaddr [lindex $argv 3] + jtag_program $filepath $serial $devaddr + } else { + jtag_program $filepath + } + } else { + error "Invalid command: $cmd" + } + disconnect_server + exit +} diff --git a/.ci/utils/mutex_hardware.py b/.ci/utils/mutex_hardware.py new file mode 100644 index 000000000..563ec5b1a --- /dev/null +++ b/.ci/utils/mutex_hardware.py @@ -0,0 +1,78 @@ +# mutex_hardware uses redis get a lock on hardware +# to prevent other Azure Pipeline agents from use. +# It also provides helper functions to get devices +# into a state where it can be used for testing. + +import argparse +import os +import pathlib +import shlex +import subprocess +import sys +import time + +from fabric import Connection +from pottery import Redlock +from redis import Redis + + +def jtag_x3xx(jtag_args, redis_server): + remote_working_dir = "pipeline_fpga" + vivado_program_jtag = "/opt/Xilinx/Vivado_Lab/2020.1/bin/vivado_lab -mode batch -source {}/viv_hardware_utils.tcl -nolog -nojournal -tclargs program".format( + remote_working_dir) + jtag_server, jtag_serial, fpga_path = jtag_args.split(",") + print("Waiting on jtag mutex for {}".format(jtag_server), flush=True) + with Redlock(key="hw_jtag_{}".format(jtag_server), + masters=redis_server, auto_release_time=1000 * 60 * 5): + print("Got jtag mutex for {}".format(jtag_server), flush=True) + with Connection(host=jtag_server) as jtag_host: + jtag_host.run("mkdir -p " + remote_working_dir) + jtag_host.run("rm -rf {}/*".format(remote_working_dir)) + jtag_host.put( + os.path.join(pathlib.Path( + __file__).parent.absolute(), "jtag/viv_hardware_utils.tcl"), + remote=remote_working_dir) + jtag_host.put(fpga_path, remote=remote_working_dir) + jtag_host.run(vivado_program_jtag + " " + + os.path.join(remote_working_dir, os.path.basename(fpga_path)) + + " " + jtag_serial) + print("Waiting 15 seconds for device to come back up", flush=True) + time.sleep(15) + + +def main(args): + redis_server = {Redis.from_url( + "redis://{}:6379/0".format(args.redis_server))} + print("Waiting to acquire mutex for {}".format(args.dut_name), flush=True) + with Redlock(key=args.dut_name, masters=redis_server, auto_release_time=1000 * 60 * args.dut_timeout): + print("Got mutex for {}".format(args.dut_name), flush=True) + if(args.jtag_x3xx != None): + jtag_x3xx(args.jtag_x3xx, redis_server) + for command in args.test_commands: + result = subprocess.run(shlex.split(command)) + if(result.returncode != 0): + sys.exit(result.returncode) + sys.exit(0) + + +if __name__ == "__main__": + parser = argparse.ArgumentParser() + # jtag_x3xx will flash the fpga for a given jtag_serial using + # Vivado on jtag_server. It uses SSH to control jtag_server. + # Provide fpga_path as a local path and it will be copied + # to jtag_server. + parser.add_argument("--jtag_x3xx", type=str, + help="user@jtag_server,jtag_serial,fpga_path") + parser.add_argument("--dut_timeout", type=int, default=30, + help="Dut mutex timeout in minutes") + parser.add_argument("redis_server", type=str, + help="Redis server for mutex") + parser.add_argument("dut_name", type=str, + help="Unique identifier for device under test") + # test_commands allows for any number of shell commands + # to execute. Call into mutex_hardware with an unlimited + # number of commands in string format as the last positional arguments. + parser.add_argument("test_commands", type=str, + nargs="+", help="Commands to run") + args = parser.parse_args() + main(args) |