[llvm] 541d6c3 - [AMDGPU][True16][CodeGen] update VGPRimm for t16 (#131021)

[via llvm-commits]

Author: Brox Chen
Date: 2025-03-20T14:46:29-04:00
New Revision: 541d6c3189cc04b5d6236972093ea3a4892a0ba8

URL: https://github.com/llvm/llvm-project/commit/541d6c3189cc04b5d6236972093ea3a4892a0ba8
DIFF: https://github.com/llvm/llvm-project/commit/541d6c3189cc04b5d6236972093ea3a4892a0ba8.diff

LOG: [AMDGPU][True16][CodeGen] update VGPRimm for t16 (#131021)

added a `bitcast_fpimm_to_i16` and update the VGPRImm pattern for t16
flow.

This change is following the pattern from the 32bit case

Added: 
    

Modified: 
    llvm/lib/Target/AMDGPU/SIInstrInfo.td
    llvm/lib/Target/AMDGPU/SIInstructions.td

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/AMDGPU/SIInstrInfo.td b/llvm/lib/Target/AMDGPU/SIInstrInfo.td
index 403c657c64053..23a7f508dcda2 100644
--- a/llvm/lib/Target/AMDGPU/SIInstrInfo.td
+++ b/llvm/lib/Target/AMDGPU/SIInstrInfo.td
@@ -846,6 +846,11 @@ def cond_as_i32imm: SDNodeXForm<cond, [{
 }]>;
 
 // Copied from the AArch64 backend:
+def bitcast_fpimm_to_i16 : SDNodeXForm<fpimm, [{
+return CurDAG->getTargetConstant(
+  N->getValueAPF().bitcastToAPInt().getZExtValue(), SDLoc(N), MVT::i16);
+}]>;
+
 def bitcast_fpimm_to_i32 : SDNodeXForm<fpimm, [{
 return CurDAG->getTargetConstant(
   N->getValueAPF().bitcastToAPInt().getZExtValue(), SDLoc(N), MVT::i32);

diff  --git a/llvm/lib/Target/AMDGPU/SIInstructions.td b/llvm/lib/Target/AMDGPU/SIInstructions.td
index 0196a5f55fb99..4b29c6a4e4751 100644
--- a/llvm/lib/Target/AMDGPU/SIInstructions.td
+++ b/llvm/lib/Target/AMDGPU/SIInstructions.td
@@ -2308,7 +2308,7 @@ let True16Predicate = UseRealTrue16Insts in {
   foreach vt = [f16, bf16] in {
     def : GCNPat <
       (VGPRImm<(vt fpimm)>:$imm),
-      (V_MOV_B16_t16_e64 0, $imm, 0)
+      (V_MOV_B16_t16_e64 0, (vt (bitcast_fpimm_to_i16 $imm)), 0)
     >;
   }
 }