1ca765153SMark Brown // SPDX-License-Identifier: GPL-2.0-only
2ca765153SMark Brown /*
3ca765153SMark Brown * Copyright (C) 2015-2020 ARM Limited.
4ca765153SMark Brown * Original author: Dave Martin <Dave.Martin@arm.com>
5ca765153SMark Brown */
6ca765153SMark Brown #include <assert.h>
7ca765153SMark Brown #include <errno.h>
8ca765153SMark Brown #include <stdio.h>
9ca765153SMark Brown #include <stdlib.h>
10ca765153SMark Brown #include <string.h>
11ca765153SMark Brown #include <sys/auxv.h>
12ca765153SMark Brown #include <sys/prctl.h>
13ca765153SMark Brown #include <asm/sigcontext.h>
14ca765153SMark Brown
15ca765153SMark Brown #include "../../kselftest.h"
16*b43ab36aSMark Brown #include "rdvl.h"
17ca765153SMark Brown
main(int argc,char ** argv)18ca765153SMark Brown int main(int argc, char **argv)
19ca765153SMark Brown {
20ca765153SMark Brown unsigned int vq;
21ca765153SMark Brown int vl;
22ca765153SMark Brown static unsigned int vqs[SVE_VQ_MAX];
23ca765153SMark Brown unsigned int nvqs = 0;
24ca765153SMark Brown
25ca765153SMark Brown ksft_print_header();
26ca765153SMark Brown ksft_set_plan(2);
27ca765153SMark Brown
28ca765153SMark Brown if (!(getauxval(AT_HWCAP) & HWCAP_SVE))
2963ebdb77SMark Brown ksft_exit_skip("SVE not available\n");
30ca765153SMark Brown
31ca765153SMark Brown /*
32ca765153SMark Brown * Enumerate up to SVE_VQ_MAX vector lengths
33ca765153SMark Brown */
34ca765153SMark Brown for (vq = SVE_VQ_MAX; vq > 0; --vq) {
35ca765153SMark Brown vl = prctl(PR_SVE_SET_VL, vq * 16);
36ca765153SMark Brown if (vl == -1)
37ca765153SMark Brown ksft_exit_fail_msg("PR_SVE_SET_VL failed: %s (%d)\n",
38ca765153SMark Brown strerror(errno), errno);
39ca765153SMark Brown
40ca765153SMark Brown vl &= PR_SVE_VL_LEN_MASK;
41ca765153SMark Brown
42*b43ab36aSMark Brown if (rdvl_sve() != vl)
43*b43ab36aSMark Brown ksft_exit_fail_msg("PR_SVE_SET_VL reports %d, RDVL %d\n",
44*b43ab36aSMark Brown vl, rdvl_sve());
45*b43ab36aSMark Brown
46ca765153SMark Brown if (!sve_vl_valid(vl))
47ca765153SMark Brown ksft_exit_fail_msg("VL %d invalid\n", vl);
48ca765153SMark Brown vq = sve_vq_from_vl(vl);
49ca765153SMark Brown
50ca765153SMark Brown if (!(nvqs < SVE_VQ_MAX))
51ca765153SMark Brown ksft_exit_fail_msg("Too many VLs %u >= SVE_VQ_MAX\n",
52ca765153SMark Brown nvqs);
53ca765153SMark Brown vqs[nvqs++] = vq;
54ca765153SMark Brown }
55ca765153SMark Brown ksft_test_result_pass("Enumerated %d vector lengths\n", nvqs);
56ca765153SMark Brown ksft_test_result_pass("All vector lengths valid\n");
57ca765153SMark Brown
58ca765153SMark Brown /* Print out the vector lengths in ascending order: */
59ca765153SMark Brown while (nvqs--)
60ca765153SMark Brown ksft_print_msg("%u\n", 16 * vqs[nvqs]);
61ca765153SMark Brown
62ca765153SMark Brown ksft_exit_pass();
63ca765153SMark Brown }
64