1 //===- GISelUtilsTest.cpp -------------------------------------------------===// 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 #include "llvm/CodeGen/GlobalISel/Utils.h" 10 #include "gtest/gtest.h" 11 12 using namespace llvm; 13 14 namespace { 15 static const LLT S1 = LLT::scalar(1); 16 static const LLT S8 = LLT::scalar(8); 17 static const LLT S16 = LLT::scalar(16); 18 static const LLT S32 = LLT::scalar(32); 19 static const LLT S64 = LLT::scalar(64); 20 static const LLT P0 = LLT::pointer(0, 64); 21 static const LLT P1 = LLT::pointer(1, 32); 22 23 static const LLT V2S8 = LLT::vector(2, 8); 24 static const LLT V4S8 = LLT::vector(4, 8); 25 26 static const LLT V2S16 = LLT::vector(2, 16); 27 static const LLT V3S16 = LLT::vector(3, 16); 28 static const LLT V4S16 = LLT::vector(4, 16); 29 30 static const LLT V2S32 = LLT::vector(2, 32); 31 static const LLT V3S32 = LLT::vector(3, 32); 32 static const LLT V4S32 = LLT::vector(4, 32); 33 static const LLT V6S32 = LLT::vector(6, 32); 34 35 static const LLT V2S64 = LLT::vector(2, 64); 36 static const LLT V4S64 = LLT::vector(4, 64); 37 38 static const LLT V2P0 = LLT::vector(2, P0); 39 static const LLT V3P0 = LLT::vector(3, P0); 40 static const LLT V4P0 = LLT::vector(4, P0); 41 static const LLT V6P0 = LLT::vector(6, P0); 42 43 static const LLT V2P1 = LLT::vector(2, P1); 44 static const LLT V4P1 = LLT::vector(4, P1); 45 46 TEST(GISelUtilsTest, getGCDType) { 47 EXPECT_EQ(S1, getGCDType(S1, S1)); 48 EXPECT_EQ(S32, getGCDType(S32, S32)); 49 EXPECT_EQ(S1, getGCDType(S1, S32)); 50 EXPECT_EQ(S1, getGCDType(S32, S1)); 51 EXPECT_EQ(S16, getGCDType(S16, S32)); 52 EXPECT_EQ(S16, getGCDType(S32, S16)); 53 54 EXPECT_EQ(V2S32, getGCDType(V2S32, V2S32)); 55 EXPECT_EQ(S32, getGCDType(V3S32, V2S32)); 56 EXPECT_EQ(S32, getGCDType(V2S32, V3S32)); 57 58 EXPECT_EQ(V2S16, getGCDType(V4S16, V2S16)); 59 EXPECT_EQ(V2S16, getGCDType(V2S16, V4S16)); 60 61 EXPECT_EQ(V2S32, getGCDType(V4S32, V2S32)); 62 EXPECT_EQ(V2S32, getGCDType(V2S32, V4S32)); 63 64 EXPECT_EQ(S16, getGCDType(P0, S16)); 65 EXPECT_EQ(S16, getGCDType(S16, P0)); 66 67 EXPECT_EQ(S32, getGCDType(P0, S32)); 68 EXPECT_EQ(S32, getGCDType(S32, P0)); 69 70 EXPECT_EQ(P0, getGCDType(P0, S64)); 71 EXPECT_EQ(S64, getGCDType(S64, P0)); 72 73 EXPECT_EQ(S32, getGCDType(P0, P1)); 74 EXPECT_EQ(S32, getGCDType(P1, P0)); 75 76 EXPECT_EQ(P0, getGCDType(V3P0, V2P0)); 77 EXPECT_EQ(P0, getGCDType(V2P0, V3P0)); 78 79 EXPECT_EQ(P0, getGCDType(P0, V2P0)); 80 EXPECT_EQ(P0, getGCDType(V2P0, P0)); 81 82 83 EXPECT_EQ(V2P0, getGCDType(V2P0, V2P0)); 84 EXPECT_EQ(P0, getGCDType(V3P0, V2P0)); 85 EXPECT_EQ(P0, getGCDType(V2P0, V3P0)); 86 EXPECT_EQ(V2P0, getGCDType(V4P0, V2P0)); 87 88 EXPECT_EQ(V2P0, getGCDType(V2P0, V4P1)); 89 EXPECT_EQ(V4P1, getGCDType(V4P1, V2P0)); 90 91 EXPECT_EQ(V2P0, getGCDType(V4P0, V4P1)); 92 EXPECT_EQ(V4P1, getGCDType(V4P1, V4P0)); 93 94 // Elements have same size, but have different pointeriness, so prefer the 95 // original element type. 96 EXPECT_EQ(V2P0, getGCDType(V2P0, V4S64)); 97 EXPECT_EQ(V2S64, getGCDType(V4S64, V2P0)); 98 99 EXPECT_EQ(V2S16, getGCDType(V2S16, V4P1)); 100 EXPECT_EQ(P1, getGCDType(V4P1, V2S16)); 101 EXPECT_EQ(V2P1, getGCDType(V4P1, V4S16)); 102 EXPECT_EQ(V4S16, getGCDType(V4S16, V2P1)); 103 104 EXPECT_EQ(P0, getGCDType(P0, V2S64)); 105 EXPECT_EQ(S64, getGCDType(V2S64, P0)); 106 107 EXPECT_EQ(S16, getGCDType(V2S16, V3S16)); 108 EXPECT_EQ(S16, getGCDType(V3S16, V2S16)); 109 EXPECT_EQ(S16, getGCDType(V3S16, S16)); 110 EXPECT_EQ(S16, getGCDType(S16, V3S16)); 111 112 EXPECT_EQ(V2S16, getGCDType(V2S16, V2S32)); 113 EXPECT_EQ(S32, getGCDType(V2S32, V2S16)); 114 115 EXPECT_EQ(V4S8, getGCDType(V4S8, V2S32)); 116 EXPECT_EQ(S32, getGCDType(V2S32, V4S8)); 117 118 // Test cases where neither element type nicely divides. 119 EXPECT_EQ(LLT::scalar(3), getGCDType(LLT::vector(3, 5), LLT::vector(2, 6))); 120 EXPECT_EQ(LLT::scalar(3), getGCDType(LLT::vector(2, 6), LLT::vector(3, 5))); 121 122 // Have to go smaller than a pointer element. 123 EXPECT_EQ(LLT::scalar(3), getGCDType(LLT::vector(2, LLT::pointer(3, 6)), 124 LLT::vector(3, 5))); 125 EXPECT_EQ(LLT::scalar(3), getGCDType(LLT::vector(3, 5), 126 LLT::vector(2, LLT::pointer(3, 6)))); 127 128 EXPECT_EQ(V4S8, getGCDType(V4S8, S32)); 129 EXPECT_EQ(S32, getGCDType(S32, V4S8)); 130 EXPECT_EQ(V4S8, getGCDType(V4S8, P1)); 131 EXPECT_EQ(P1, getGCDType(P1, V4S8)); 132 133 EXPECT_EQ(V2S8, getGCDType(V2S8, V4S16)); 134 EXPECT_EQ(S16, getGCDType(V4S16, V2S8)); 135 136 EXPECT_EQ(S8, getGCDType(V2S8, LLT::vector(4, 2))); 137 EXPECT_EQ(LLT::vector(4, 2), getGCDType(LLT::vector(4, 2), S8)); 138 139 140 EXPECT_EQ(LLT::pointer(4, 8), getGCDType(LLT::vector(2, LLT::pointer(4, 8)), 141 LLT::vector(4, 2))); 142 143 EXPECT_EQ(LLT::vector(4, 2), getGCDType(LLT::vector(4, 2), 144 LLT::vector(2, LLT::pointer(4, 8)))); 145 146 EXPECT_EQ(LLT::scalar(4), getGCDType(LLT::vector(3, 4), S8)); 147 EXPECT_EQ(LLT::scalar(4), getGCDType(S8, LLT::vector(3, 4))); 148 } 149 150 TEST(GISelUtilsTest, getLCMType) { 151 EXPECT_EQ(S1, getLCMType(S1, S1)); 152 EXPECT_EQ(S32, getLCMType(S32, S1)); 153 EXPECT_EQ(S32, getLCMType(S1, S32)); 154 EXPECT_EQ(S32, getLCMType(S32, S32)); 155 156 EXPECT_EQ(S32, getLCMType(S32, S16)); 157 EXPECT_EQ(S32, getLCMType(S16, S32)); 158 159 EXPECT_EQ(S64, getLCMType(S64, P0)); 160 EXPECT_EQ(S64, getLCMType(P0, S64)); 161 162 EXPECT_EQ(S64, getLCMType(S32, P0)); 163 EXPECT_EQ(S64, getLCMType(P0, S32)); 164 165 EXPECT_EQ(S32, getLCMType(S32, P1)); 166 EXPECT_EQ(S32, getLCMType(P1, S32)); 167 EXPECT_EQ(S64, getLCMType(P0, P0)); 168 EXPECT_EQ(S32, getLCMType(P1, P1)); 169 170 EXPECT_EQ(S64, getLCMType(P0, P1)); 171 EXPECT_EQ(S64, getLCMType(P1, P0)); 172 173 EXPECT_EQ(V2S32, getLCMType(V2S32, V2S32)); 174 EXPECT_EQ(V2S32, getLCMType(V2S32, S32)); 175 EXPECT_EQ(V2S32, getLCMType(S32, V2S32)); 176 EXPECT_EQ(V2S32, getLCMType(V2S32, V2S32)); 177 EXPECT_EQ(V6S32, getLCMType(V2S32, V3S32)); 178 EXPECT_EQ(V6S32, getLCMType(V3S32, V2S32)); 179 EXPECT_EQ(LLT::vector(12, S32), getLCMType(V4S32, V3S32)); 180 EXPECT_EQ(LLT::vector(12, S32), getLCMType(V3S32, V4S32)); 181 182 EXPECT_EQ(V2P0, getLCMType(V2P0, V2P0)); 183 EXPECT_EQ(V2P0, getLCMType(V2P0, P0)); 184 EXPECT_EQ(V2P0, getLCMType(P0, V2P0)); 185 EXPECT_EQ(V2P0, getLCMType(V2P0, V2P0)); 186 EXPECT_EQ(V6P0, getLCMType(V2P0, V3P0)); 187 EXPECT_EQ(V6P0, getLCMType(V3P0, V2P0)); 188 EXPECT_EQ(LLT::vector(12, P0), getLCMType(V4P0, V3P0)); 189 EXPECT_EQ(LLT::vector(12, P0), getLCMType(V3P0, V4P0)); 190 191 // FIXME 192 // EXPECT_EQ(V2S32, getLCMType(V2S32, S64)); 193 194 // FIXME 195 //EXPECT_EQ(S64, getLCMType(S64, V2S32)); 196 } 197 198 } 199