[llvm] [AMDGPU] select v_sat_pk from two i16 or v2i16 (PR #121124)

via llvm-commits llvm-commits at lists.llvm.org
Wed Jan 15 02:16:55 PST 2025


================
@@ -3298,6 +3301,34 @@ def : GCNPat <
   (v2i16 (V_LSHL_OR_B32_e64 $src1, (i32 16), (i32 (V_AND_B32_e64 (i32 (V_MOV_B32_e32 (i32 0xffff))), $src0))))
 >;
 
+multiclass V_SAT_PK_Pat<Instruction inst> {
+  def: GCNPatIgnoreCopies<
+    (i16 (conc_lo_u8_i16 (clamp_s16_u8 i16:$lo), (clamp_s16_u8 i16:$hi))),
+    (inst
+      (V_LSHL_OR_B32_e64 VRegSrc_32:$hi, (S_MOV_B32 (i32 16)),
+      VRegSrc_32:$lo))
+  >;
+
+  def: GCNPatIgnoreCopies<
+    (i16 (conc_lo_u8_i16 (clamp_s16_u8 i16:$lo), (smax i16:$hi, (i16 0)))),
----------------
Shoreshen wrote:

Hi @arsenm , may I ask why since if there are other uses of the `(clamp_s16_u8 i16:$lo)`, the DAG is just not going to fold...

https://github.com/llvm/llvm-project/pull/121124


More information about the llvm-commits mailing list