| 1 | # |
| 2 | # Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries. |
| 3 | # |
| 4 | # SPDX-License-Identifier: GPL-2.0-or-later |
| 5 | |
| 6 | # |
| 7 | # Test the SME registers are visible and changeable via gdbstub |
| 8 | # |
| 9 | # This is launched via tests/guest-debug/run-test.py |
| 10 | # |
| 11 | |
| 12 | import argparse |
| 13 | import gdb |
| 14 | from test_gdbstub import main, report |
| 15 | |
| 16 | MAGIC = 0x01020304 |
| 17 | BASIC_ZA_TEST = 0 |
| 18 | TILE_SLICE_TEST = 0 |
| 19 | |
| 20 | |
| 21 | def run_test(): |
| 22 | """Run the requested test(s) for SME ZA gdbstub support""" |
| 23 | |
| 24 | if BASIC_ZA_TEST: |
| 25 | run_basic_sme_za_gdbstub_support_test() |
| 26 | if TILE_SLICE_TEST: |
| 27 | run_basic_sme_za_tile_slice_gdbstub_support_test() |
| 28 | |
| 29 | |
| 30 | def run_basic_sme_za_gdbstub_support_test(): |
| 31 | """Test reads and writes to the SME ZA register at the byte level""" |
| 32 | |
| 33 | frame = gdb.selected_frame() |
| 34 | rname = "za" |
| 35 | za = frame.read_register(rname) |
| 36 | report(True, "Reading %s" % rname) |
| 37 | |
| 38 | # Writing to the ZA register, byte by byte. |
| 39 | for i in range(0, 16): |
| 40 | for j in range(0, 16): |
| 41 | cmd = "set $za[%d][%d] = 0x01" % (i, j) |
| 42 | gdb.execute(cmd) |
| 43 | report(True, "%s" % cmd) |
| 44 | |
| 45 | # Reading from the ZA register, byte by byte. |
| 46 | for i in range(0, 16): |
| 47 | for j in range(0, 16): |
| 48 | reg = "$za[%d][%d]" % (i, j) |
| 49 | v = gdb.parse_and_eval(reg) |
| 50 | report(str(v.type) == "uint8_t", "size of %s" % (reg)) |
| 51 | report(v == 0x1, "%s is 0x%x" % (reg, 0x1)) |
| 52 | |
| 53 | |
| 54 | def run_basic_sme_za_tile_slice_gdbstub_support_test(): |
| 55 | """Test reads and writes of SME ZA horizontal and vertical tile slices |
| 56 | |
| 57 | Test if SME ZA tile slices, both horizontal and vertical, |
| 58 | can be correctly read and written to. The sizes to test |
| 59 | are quadwords and doublewords. |
| 60 | """ |
| 61 | |
| 62 | sizes = {} |
| 63 | sizes["q"] = "uint128_t" |
| 64 | sizes["d"] = "uint64_t" |
| 65 | |
| 66 | # Accessing requested sizes of elements of ZA |
| 67 | for size in sizes: |
| 68 | |
| 69 | # Accessing various ZA tiles |
| 70 | for i in range(0, 4): |
| 71 | |
| 72 | # Accessing various horizontal slices for each ZA tile |
| 73 | for j in range(0, 4): |
| 74 | # Writing to various elements in each tile slice |
| 75 | for k in range(0, 4): |
| 76 | cmd = "set $za%dh%c%d[%d] = 0x%x" % (i, size, j, k, MAGIC) |
| 77 | gdb.execute(cmd) |
| 78 | report(True, "%s" % cmd) |
| 79 | |
| 80 | # Reading from the written elements in each tile slice |
| 81 | for k in range(0, 4): |
| 82 | reg = "$za%dh%c%d[%d]" % (i, size, j, k) |
| 83 | v = gdb.parse_and_eval(reg) |
| 84 | report(str(v.type) == sizes[size], "size of %s" % (reg)) |
| 85 | report(v == MAGIC, "%s is 0x%x" % (reg, MAGIC)) |
| 86 | |
| 87 | # Accessing various vertical slices for each ZA tile |
| 88 | for j in range(0, 4): |
| 89 | # Writing to various elements in each tile slice |
| 90 | for k in range(0, 4): |
| 91 | cmd = "set $za%dv%c%d[%d] = 0x%x" % (i, size, j, k, MAGIC) |
| 92 | gdb.execute(cmd) |
| 93 | report(True, "%s" % cmd) |
| 94 | |
| 95 | # Reading from the written elements in each tile slice |
| 96 | for k in range(0, 4): |
| 97 | reg = "$za%dv%c%d[%d]" % (i, size, j, k) |
| 98 | v = gdb.parse_and_eval(reg) |
| 99 | report(str(v.type) == sizes[size], "size of %s" % (reg)) |
| 100 | report(v == MAGIC, "%s is 0x%x" % (reg, MAGIC)) |
| 101 | |
| 102 | |
| 103 | parser = argparse.ArgumentParser(description="A gdbstub test for SME support") |
| 104 | parser.add_argument("--gdb_basic_za_test", |
| 105 | help="Enable test for basic SME ZA support", |
| 106 | action="store_true") |
| 107 | parser.add_argument("--gdb_tile_slice_test", |
| 108 | help="Enable test for ZA tile slice support", |
| 109 | action="store_true") |
| 110 | args = parser.parse_args() |
| 111 | |
| 112 | if args.gdb_basic_za_test: |
| 113 | BASIC_ZA_TEST = 1 |
| 114 | if args.gdb_tile_slice_test: |
| 115 | TILE_SLICE_TEST = 1 |
| 116 | |
| 117 | main(run_test, expected_arch="aarch64") |