1 // SPDX-License-Identifier: GPL-2.0-only 2 // Copyright (C) 2021 ARM Limited. 3 4 #include "sme-inst.h" 5 6 .arch_extension sve 7 8 #define MAGIC 42 9 10 #define MAXVL 2048 11 #define MAXVL_B (MAXVL / 8) 12 13 .pushsection .text 14 .data 15 .align 4 16 scratch: 17 .space MAXVL_B 18 .popsection 19 20 .globl fork_test 21 fork_test: 22 smstart_za 23 24 // For simplicity just set one word in one vector, other tests 25 // cover general data corruption issues. 26 ldr x0, =scratch 27 mov x1, #MAGIC 28 str x1, [x0] 29 mov w12, wzr 30 _ldr_za 12, 0 // ZA.H[W12] loaded from [X0] 31 32 // Tail call into the C portion that does the fork & verify 33 b fork_test_c 34 35 .globl verify_fork 36 verify_fork: 37 // SVCR should have ZA=1, SM=0 38 mrs x0, S3_3_C4_C2_2 39 and x1, x0, #3 40 cmp x1, #2 41 beq 1f 42 mov x0, xzr 43 b 100f 44 1: 45 46 // ZA should still have the value we loaded 47 ldr x0, =scratch 48 mov w12, wzr 49 _str_za 12, 0 // ZA.H[W12] stored to [X0] 50 ldr x1, [x0] 51 cmp x1, #MAGIC 52 beq 2f 53 mov x0, xzr 54 b 100f 55 56 2: 57 // All tests passed 58 mov x0, #1 59 100: 60 ret 61 62