emit flags

Author: shiltian

llvm/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp

@@ -5919,11 +5919,15 @@ bool AMDGPULegalizerInfo::legalizePointerAsRsrcIntrin(
     Register LowHalf = B.buildOr(S64, ExtPointer, NumRecordsLHS).getReg(0);
 
     // Build the higher 64-bit value, which has the higher 38-bit num_records,
-    // 6-bit zero (omit), 14-bit stride and 6-bit zero (omit).
+    // 6-bit zero (omit), 16-bit stride and scale and 4-bit flag.
     auto NumRecordsRHS = B.buildLShr(S64, NumRecords, B.buildConstant(S32, 7));
     auto ExtStride = B.buildAnyExt(S64, Stride);
     auto ShiftedStride = B.buildShl(S64, ExtStride, B.buildConstant(S32, 44));
-    Register HighHalf = B.buildOr(S64, NumRecordsRHS, ShiftedStride).getReg(0);
+    auto ExtFlags = B.buildAnyExt(S64, Flags);
+    auto NewFlags = B.buildAnd(S64, ExtFlags, B.buildConstant(S64, 0x3));
+    auto ShiftedFlags = B.buildShl(S64, NewFlags, B.buildConstant(S32, 60));
+    auto CombinedFields = B.buildOr(S64, NumRecordsRHS, ShiftedStride);
+    Register HighHalf = B.buildOr(S64, CombinedFields, ShiftedFlags).getReg(0);
     B.buildMergeValues(Result, {LowHalf, HighHalf});
   } else {
     NumRecords = B.buildTrunc(S32, NumRecords).getReg(0);

llvm/lib/Target/AMDGPU/SIISelLowering.cpp

@@ -11615,16 +11615,24 @@ SDValue SITargetLowering::lowerPointerAsRsrcIntrin(SDNode *Op,
         DAG.getNode(ISD::OR, Loc, MVT::i64, ExtPointer, NumRecordsLHS);
 
     // Build the higher 64-bit value, which has the higher 38-bit num_records,
-    // 6-bit zero (omit), 14-bit stride and 6-bit zero (omit).
+    // 6-bit zero (omit), 16-bit stride and scale and 4-bit flag.
     SDValue NumRecordsRHS =
         DAG.getNode(ISD::SRL, Loc, MVT::i64, NumRecords,
                     DAG.getShiftAmountConstant(7, MVT::i32, Loc));
     SDValue ExtStride = DAG.getAnyExtOrTrunc(Stride, Loc, MVT::i64);
     SDValue ShiftedStride =
         DAG.getNode(ISD::SHL, Loc, MVT::i64, ExtStride,
                     DAG.getShiftAmountConstant(44, MVT::i32, Loc));
-    SDValue HighHalf =
+    SDValue ExtFlags = DAG.getAnyExtOrTrunc(Flags, Loc, MVT::i64);
+    SDValue NewFlags = DAG.getNode(ISD::AND, Loc, MVT::i64, ExtFlags,
+                                   DAG.getConstant(0x3, Loc, MVT::i64));
+    SDValue ShiftedFlags =
+        DAG.getNode(ISD::SHL, Loc, MVT::i64, NewFlags,
+                    DAG.getShiftAmountConstant(60, MVT::i32, Loc));
+    SDValue CombinedFields =
         DAG.getNode(ISD::OR, Loc, MVT::i64, NumRecordsRHS, ShiftedStride);
+    SDValue HighHalf =
+        DAG.getNode(ISD::OR, Loc, MVT::i64, CombinedFields, ShiftedFlags);
 
     Rsrc = DAG.getNode(ISD::BUILD_VECTOR, Loc, MVT::v2i64, LowHalf, HighHalf);
   } else {