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
changeTo(unsigned TypeIdx,LLT Ty)17 LegalizeMutation LegalizeMutations::changeTo(unsigned TypeIdx, LLT Ty) {
18 return
19 [=](const LegalityQuery &Query) { return std::make_pair(TypeIdx, Ty); };
20 }
21
changeTo(unsigned TypeIdx,unsigned FromTypeIdx)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
changeElementTo(unsigned TypeIdx,unsigned FromTypeIdx)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
changeElementTo(unsigned TypeIdx,LLT NewEltTy)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
widenScalarOrEltToNextPow2(unsigned TypeIdx,unsigned Min)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
moreElementsToNextPow2(unsigned TypeIdx,unsigned Min)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
scalarize(unsigned TypeIdx)67 LegalizeMutation LegalizeMutations::scalarize(unsigned TypeIdx) {
68 return [=](const LegalityQuery &Query) {
69 return std::make_pair(TypeIdx, Query.Types[TypeIdx].getElementType());
70 };
71 }
72