contrib/llvm-project/llvm/lib/CodeGen/GlobalISel/LegalizeMutations.cpp

   1 //===- lib/CodeGen/GlobalISel/LegalizerMutations.cpp - Mutations ----------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // A library of mutation factories to use for LegalityMutation.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "llvm/CodeGen/GlobalISel/LegalizerInfo.h"
  14
  15 using namespace llvm;
  16
  17 LegalizeMutation LegalizeMutations::changeTo(unsigned TypeIdx, LLT Ty) {
  18   return
  19       [=](const LegalityQuery &Query) { return std::make_pair(TypeIdx, Ty); };
  20 }
  21
  22 LegalizeMutation LegalizeMutations::changeTo(unsigned TypeIdx,
  23                                              unsigned FromTypeIdx) {
  24   return [=](const LegalityQuery &Query) {
  25     return std::make_pair(TypeIdx, Query.Types[FromTypeIdx]);
  26   };
  27 }
  28
  29 LegalizeMutation LegalizeMutations::changeElementTo(unsigned TypeIdx,
  30                                                     unsigned FromTypeIdx) {
  31   return [=](const LegalityQuery &Query) {
  32     const LLT OldTy = Query.Types[TypeIdx];
  33     const LLT NewTy = Query.Types[FromTypeIdx];
  34     return std::make_pair(TypeIdx, OldTy.changeElementType(NewTy));
  35   };
  36 }
  37
  38 LegalizeMutation LegalizeMutations::changeElementTo(unsigned TypeIdx,
  39                                                     LLT NewEltTy) {
  40   return [=](const LegalityQuery &Query) {
  41     const LLT OldTy = Query.Types[TypeIdx];
  42     return std::make_pair(TypeIdx, OldTy.changeElementType(NewEltTy));
  43   };
  44 }
  45
  46 LegalizeMutation LegalizeMutations::widenScalarOrEltToNextPow2(unsigned TypeIdx,
  47                                                                unsigned Min) {
  48   return [=](const LegalityQuery &Query) {
  49     const LLT Ty = Query.Types[TypeIdx];
  50     unsigned NewEltSizeInBits =
  51         std::max(1u << Log2_32_Ceil(Ty.getScalarSizeInBits()), Min);
  52     return std::make_pair(TypeIdx, Ty.changeElementSize(NewEltSizeInBits));
  53   };
  54 }
  55
  56 LegalizeMutation LegalizeMutations::moreElementsToNextPow2(unsigned TypeIdx,
  57                                                            unsigned Min) {
  58   return [=](const LegalityQuery &Query) {
  59     const LLT VecTy = Query.Types[TypeIdx];
  60     unsigned NewNumElements =
  61         std::max(1u << Log2_32_Ceil(VecTy.getNumElements()), Min);
  62     return std::make_pair(TypeIdx,
  63                           LLT::vector(NewNumElements, VecTy.getElementType()));
  64   };
  65 }
  66
  67 LegalizeMutation LegalizeMutations::scalarize(unsigned TypeIdx) {
  68   return [=](const LegalityQuery &Query) {
  69     return std::make_pair(TypeIdx, Query.Types[TypeIdx].getElementType());
  70   };
  71 }