diff options
| author | Jessica Del <50999226+OutOfCache@users.noreply.github.com> | 2023-11-21 16:36:45 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-21 16:36:45 +0100 |
| commit | f85e7ab03534842d57d8fd385617a8d56b31bf09 (patch) | |
| tree | 0f4513912d13780ebd8c8aa344a6eab90d3e733a /llvm/lib/Analysis/ConstantFolding.cpp | |
| parent | f33588380886e1bda124b213a90b7c8163ca3d3b (diff) | |
| download | llvm-f85e7ab03534842d57d8fd385617a8d56b31bf09.zip llvm-f85e7ab03534842d57d8fd385617a8d56b31bf09.tar.gz llvm-f85e7ab03534842d57d8fd385617a8d56b31bf09.tar.bz2 | |
[AMDGPU] - Add constant folding to s_wqm intrinsic (#72382)
Fold any constant input to the `s_wqm` intrinsic.
Diffstat (limited to 'llvm/lib/Analysis/ConstantFolding.cpp')
| -rw-r--r-- | llvm/lib/Analysis/ConstantFolding.cpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/llvm/lib/Analysis/ConstantFolding.cpp b/llvm/lib/Analysis/ConstantFolding.cpp index e8ee672b..90da339 100644 --- a/llvm/lib/Analysis/ConstantFolding.cpp +++ b/llvm/lib/Analysis/ConstantFolding.cpp @@ -1533,6 +1533,7 @@ bool llvm::canConstantFoldCallTo(const CallBase *Call, const Function *F) { case Intrinsic::amdgcn_perm: case Intrinsic::amdgcn_wave_reduce_umin: case Intrinsic::amdgcn_wave_reduce_umax: + case Intrinsic::amdgcn_s_wqm: case Intrinsic::amdgcn_s_quadmask: case Intrinsic::amdgcn_s_bitreplicate: case Intrinsic::arm_mve_vctp8: @@ -2425,6 +2426,15 @@ static Constant *ConstantFoldScalarCall1(StringRef Name, return ConstantFP::get(Ty->getContext(), Val); } + case Intrinsic::amdgcn_s_wqm: { + uint64_t Val = Op->getZExtValue(); + Val |= (Val & 0x5555555555555555ULL) << 1 | + ((Val >> 1) & 0x5555555555555555ULL); + Val |= (Val & 0x3333333333333333ULL) << 2 | + ((Val >> 2) & 0x3333333333333333ULL); + return ConstantInt::get(Ty, Val); + } + case Intrinsic::amdgcn_s_quadmask: { uint64_t Val = Op->getZExtValue(); uint64_t QuadMask = 0; |