LLVM项目中AMDGPU指令语法详解

LLVM项目中AMDGPU指令语法详解

llvm Project moved to: https://github.com/llvm/llvm-project 项目地址: https://gitcode.com/gh_mirrors/ll/llvm

概述

在LLVM编译器框架中，AMDGPU后端支持多种GPU架构的代码生成。理解AMDGPU指令语法对于编写高效的GPU代码至关重要。本文将深入解析AMDGPU指令的结构、类型后缀、编码后缀等关键概念，帮助开发者更好地理解和使用这些指令。

指令基本结构

AMDGPU指令遵循特定的语法格式，主要由以下几个部分组成：

<操作码助记符> <操作数0>, <操作数1>,... <修饰符0> <修饰符1>...

其中：

• 操作码助记符：描述指令的基本功能和数据类型
• 操作数：指令处理的数据，用逗号分隔
• 修饰符：影响指令行为的可选参数，用空格分隔

操作码助记符详解

操作码助记符不仅包含基本操作信息，还可能包含多个后缀，按特定顺序排列：

1. 目标操作数类型后缀
2. 源操作数类型后缀
3. 编码后缀

数据类型后缀

AMDGPU指令操作的数据类型通过后缀明确指定。有些指令有两个类型后缀：第一个指定目标操作数类型，第二个指定源操作数类型。

常见的数据类型后缀包括：

| 后缀 | 是否打包指令 | 数据类型 | |------|--------------|----------| | _b512, _b256等 | 否 | 位数据 | | _u64, _u32等 | 否 | 无符号整数 | | _i64, _i32等 | 否 | 有符号整数 | | _f64, _f32等 | 否 | 浮点数 | | _b16, _u16等 | 是 | 打包数据 |

大小后缀

对于无类型数据操作的指令，数据大小通过特定后缀指定：

| 后缀 | 隐含数据类型 | 所需寄存器大小(dwords) | |------|--------------|------------------------| | - | b32 | 1 | | x2 | b64 | 2 | | x3 | b96 | 3 | | x4 | b128 | 4 | | x8 | b256 | 8 | | x16 | b512 | 16 |

特殊的大小后缀如d16_x、d16_xy等用于处理16位数据，在不同GPU架构(GFX8.0、GFX8.1和GFX9)上有不同的寄存器占用情况。

编码后缀

AMDGPU指令有多种编码格式，可以通过后缀强制指定：

| 编码格式 | 后缀 | |----------|------| | 原生32位编码(VOP1/VOP2/VOPC) | _e32 | | VOP3(64位)编码 | _e64 | | DPP编码 | _dpp | | SDWA编码 | _sdwa |

如果不指定编码后缀，汇编器会根据操作数自动选择最优编码。

操作数与修饰符

操作数语法

AMDGPU指令的操作数遵循特定语法规则，主要包括：

• 寄存器操作数
• 立即数
• 特殊功能寄存器
• 内存地址等

不同类型的操作数有各自的表示方法和限制条件。

修饰符语法

修饰符用于改变指令的默认行为，常见的修饰符包括：

• 饱和修饰符
• 舍入模式修饰符
• 条件执行修饰符等

修饰符通常紧跟在操作数之后，用空格分隔。

实际应用示例

不同类型操作数的指令

s_bcnt0_i32_b64    // 目标操作数为i32，源操作数为b64
v_cvt_f32_u32      // 目标操作数为f32，源操作数为u32

单一数据类型的指令

v_max3_f32         // 所有操作数都是f32类型
v_max3_i16         // 所有操作数都是i16类型

打包数据操作指令

v_pk_add_u16       // 打包无符号16位整数加法
v_pk_add_f16       // 打包16位浮点数加法

无类型数据操作指令

buffer_store_dwordx4  // 操作b128类型数据
flat_load_dwordx4     // 操作b128类型数据

注意事项

1. 上述规则存在一些例外情况，特定指令可能有特殊的行为
2. 不同GPU架构(GFX7/8/9/10)支持的指令和修饰符可能有所不同
3. 16位数据操作在不同架构上的寄存器占用情况不同
4. 汇编器会自动选择最优编码，但可以通过后缀强制指定

理解AMDGPU指令语法对于编写高效的GPU代码至关重要。通过合理使用数据类型后缀、编码后缀和修饰符，可以充分发挥GPU的计算能力。

llvm Project moved to: https://github.com/llvm/llvm-project 项目地址: https://gitcode.com/gh_mirrors/ll/llvm