Apple Silicon(M2) Neon SIMD累加ARM64汇编程序异常排查

问题描述 #

在M2机器上编写汇编程序，目标是用Neon SIMD指令累加内置数值并输出结果（预期输出10），但实际结果不符。代码及汇编链接命令如下：

原代码 #

.global _start
.align 2

_start:
    ;;; Load numbers into x0
    ldr x0, numbers

    ;;; Load elements from array in x0 into dedicated Neon register
    ld1 { v0. 4s }, [x0]

    ;;; Accumulate elements in vector using dedicated Neon instruction
    addv s0, v0.4s 

    ;;; Prepare formatted string
    adrp    x0, format@page
    add x0, x0, format@pageoff

    ;;; Add result to the stack for printing
    str s0, [sp, #-16]!

    ;;; Print string
    bl _printf
    mov x16, #1
    svc 0

numbers: .word 1, 2, 3, 4
format: .asciz "Answer: %u.
"

汇编链接命令 #

as -g -arch arm64 -o test.o test.s
ld -o test test.o -lSystem -syslibroot `xcrun -sdk macosx --show-sdk-path` -e _start -arch arm64

错误点分析 #

• Neon指令语法错误：ld1 { v0. 4s }, [x0]中v0.与4s之间多了空格，正确写法应为v0.4s，空格会导致汇编指令解析失败。
• 数据类型与格式不匹配：
  • numbers定义的是32位整数（.word），但addv s0, v0.4s是按单精度浮点数规则累加，会得到错误的浮点值；同时输出格式用%u（无符号整数），与浮点类型不匹配，导致打印结果异常。
• 参数传递逻辑错误：
  • ARM64架构下printf的参数通过寄存器传递，第一个参数（格式串）存在x0，第二个参数需放在x1，而非压栈传递。浮点寄存器s0的值不能直接作为整数参数传递，需先转换为通用整数寄存器。
• 字符串格式错误：.asciz "Answer: %u. "中的换行需用\n表示，否则字符串包含无效字符，会干扰打印行为。
• 程序退出不规范：调用printf后直接用svc 0退出未返回正确状态码，macOS下应返回0表示正常退出。

修正后的代码 #

.global _start
.align 2

_start:
    ; 加载数组地址到x0
    ldr x0, numbers
    ; 加载4个32位整数到Neon寄存器v0
    ld1 { v0.4s }, [x0]
    ; 累加32位整数，结果存入通用整数寄存器w1
    addv w1, v0.4s

    ; 准备格式字符串地址到x0
    adrp x0, format@page
    add x0, x0, format@pageoff

    ; 调用printf打印结果
    bl _printf

    ; 正常退出程序，返回状态码0
    mov x0, #0
    mov x16, #1
    svc 0

numbers: .word 1, 2, 3, 4
format: .asciz "Answer: %u\n"

验证 #

重新执行汇编链接命令后运行程序，会输出Answer: 10，符合预期。

内容的提问来源于stack exchange，提问作者ironman