From 70e2d631e178fbf5cfcd45d8af43bfcad24db744 Mon Sep 17 00:00:00 2001 From: Jay Foad Date: Wed, 29 Mar 2023 13:10:19 +0100 Subject: [PATCH] [AMDGPU] Consistently use v2i16 for packed bf16 inputs. NFC. Differential Revision: https://reviews.llvm.org/D147145 --- llvm/lib/Target/AMDGPU/FLATInstructions.td | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/llvm/lib/Target/AMDGPU/FLATInstructions.td b/llvm/lib/Target/AMDGPU/FLATInstructions.td index 2782fac..5c86d80 100644 --- a/llvm/lib/Target/AMDGPU/FLATInstructions.td +++ b/llvm/lib/Target/AMDGPU/FLATInstructions.td @@ -732,11 +732,12 @@ let SubtargetPredicate = isGFX90APlus in { let SubtargetPredicate = HasAtomicFlatPkAdd16Insts in { defm FLAT_ATOMIC_PK_ADD_F16 : FLAT_Atomic_Pseudo<"flat_atomic_pk_add_f16", VGPR_32, v2f16>; - defm FLAT_ATOMIC_PK_ADD_BF16 : FLAT_Atomic_Pseudo<"flat_atomic_pk_add_bf16", VGPR_32, v2f16>; + let FPAtomic = 1 in + defm FLAT_ATOMIC_PK_ADD_BF16 : FLAT_Atomic_Pseudo<"flat_atomic_pk_add_bf16", VGPR_32, v2i16>; } // End SubtargetPredicate = HasAtomicFlatPkAdd16Insts -let SubtargetPredicate = HasAtomicGlobalPkAddBF16Inst in - defm GLOBAL_ATOMIC_PK_ADD_BF16 : FLAT_Global_Atomic_Pseudo<"global_atomic_pk_add_bf16", VGPR_32, v2f16>; +let SubtargetPredicate = HasAtomicGlobalPkAddBF16Inst, FPAtomic = 1 in + defm GLOBAL_ATOMIC_PK_ADD_BF16 : FLAT_Global_Atomic_Pseudo<"global_atomic_pk_add_bf16", VGPR_32, v2i16>; // GFX7-, GFX10-, GFX11-only flat instructions. let SubtargetPredicate = isGFX7GFX10GFX11 in { -- 2.7.4