1; RUN: mlir-translate -import-llvm %s | FileCheck %s 2 3%struct.t = type {} 4%struct.s = type { %struct.t, i64 } 5 6; CHECK: llvm.mlir.global external @g1() {alignment = 8 : i64} : !llvm.struct<"struct.s", (struct<"struct.t", ()>, i64)> 7@g1 = external global %struct.s, align 8 8; CHECK: llvm.mlir.global external @g2() {alignment = 8 : i64} : f64 9@g2 = external global double, align 8 10; CHECK: llvm.mlir.global internal @g3("string") 11@g3 = internal global [6 x i8] c"string" 12 13; CHECK: llvm.mlir.global external @g5() : vector<8xi32> 14@g5 = external global <8 x i32> 15 16; CHECK: llvm.mlir.global private @alig32(42 : i64) {alignment = 32 : i64, dso_local} : i64 17@alig32 = private global i64 42, align 32 18 19; CHECK: llvm.mlir.global private @alig64(42 : i64) {alignment = 64 : i64, dso_local} : i64 20@alig64 = private global i64 42, align 64 21 22@g4 = external global i32, align 8 23; CHECK: llvm.mlir.global internal constant @int_gep() {dso_local} : !llvm.ptr<i32> { 24; CHECK-DAG: %[[addr:[0-9]+]] = llvm.mlir.addressof @g4 : !llvm.ptr<i32> 25; CHECK-DAG: %[[c2:[0-9]+]] = llvm.mlir.constant(2 : i32) : i32 26; CHECK-NEXT: %[[gepinit:[0-9]+]] = llvm.getelementptr %[[addr]][%[[c2]]] : (!llvm.ptr<i32>, i32) -> !llvm.ptr<i32> 27; CHECK-NEXT: llvm.return %[[gepinit]] : !llvm.ptr<i32> 28; CHECK-NEXT: } 29@int_gep = internal constant i32* getelementptr (i32, i32* @g4, i32 2) 30 31; 32; dso_local attribute 33; 34 35; CHECK: llvm.mlir.global external @dso_local_var() {dso_local} : !llvm.struct<"struct.s", (struct<"struct.t", ()>, i64)> 36@dso_local_var = external dso_local global %struct.s 37 38; 39; thread_local attribute 40; 41 42; CHECK: llvm.mlir.global external thread_local @thread_local_var() : !llvm.struct<"struct.s", (struct<"struct.t", ()>, i64)> 43@thread_local_var = external thread_local global %struct.s 44 45; 46; addr_space attribute 47; 48 49; CHECK: llvm.mlir.global external @addr_space_var(0 : i32) {addr_space = 6 : i32} : i32 50@addr_space_var = addrspace(6) global i32 0 51 52; 53; Linkage attribute. 54; 55 56; CHECK: llvm.mlir.global private @private(42 : i32) {dso_local} : i32 57@private = private global i32 42 58; CHECK: llvm.mlir.global internal @internal(42 : i32) {dso_local} : i32 59@internal = internal global i32 42 60; CHECK: llvm.mlir.global available_externally @available_externally(42 : i32) : i32 61@available_externally = available_externally global i32 42 62; CHECK: llvm.mlir.global linkonce @linkonce(42 : i32) : i32 63@linkonce = linkonce global i32 42 64; CHECK: llvm.mlir.global weak @weak(42 : i32) : i32 65@weak = weak global i32 42 66; CHECK: llvm.mlir.global common @common(0 : i32) : i32 67@common = common global i32 zeroinitializer 68; CHECK: llvm.mlir.global appending @appending(dense<[0, 1]> : tensor<2xi32>) : !llvm.array<2 x i32> 69@appending = appending global [2 x i32] [i32 0, i32 1] 70; CHECK: llvm.mlir.global extern_weak @extern_weak() : i32 71@extern_weak = extern_weak global i32 72; CHECK: llvm.mlir.global linkonce_odr @linkonce_odr(42 : i32) : i32 73@linkonce_odr = linkonce_odr global i32 42 74; CHECK: llvm.mlir.global weak_odr @weak_odr(42 : i32) : i32 75@weak_odr = weak_odr global i32 42 76; CHECK: llvm.mlir.global external @external() : i32 77@external = external global i32 78 79; 80; UnnamedAddr attribute. 81; 82 83 84; CHECK: llvm.mlir.global private constant @no_unnamed_addr(42 : i64) {dso_local} : i64 85@no_unnamed_addr = private constant i64 42 86; CHECK: llvm.mlir.global private local_unnamed_addr constant @local_unnamed_addr(42 : i64) {dso_local} : i64 87@local_unnamed_addr = private local_unnamed_addr constant i64 42 88; CHECK: llvm.mlir.global private unnamed_addr constant @unnamed_addr(42 : i64) {dso_local} : i64 89@unnamed_addr = private unnamed_addr constant i64 42 90 91; 92; Section attribute 93; 94 95; CHECK: llvm.mlir.global internal constant @sectionvar("teststring") {dso_local, section = ".mysection"} 96@sectionvar = internal constant [10 x i8] c"teststring", section ".mysection" 97 98; 99; Sequential constants. 100; 101 102; CHECK: llvm.mlir.global internal constant @vector_constant(dense<[1, 2]> : vector<2xi32>) {dso_local} : vector<2xi32> 103@vector_constant = internal constant <2 x i32> <i32 1, i32 2> 104; CHECK: llvm.mlir.global internal constant @array_constant(dense<[1.000000e+00, 2.000000e+00]> : tensor<2xf32>) {dso_local} : !llvm.array<2 x f32> 105@array_constant = internal constant [2 x float] [float 1., float 2.] 106; CHECK: llvm.mlir.global internal constant @nested_array_constant(dense<[{{\[}}1, 2], [3, 4]]> : tensor<2x2xi32>) {dso_local} : !llvm.array<2 x array<2 x i32>> 107@nested_array_constant = internal constant [2 x [2 x i32]] [[2 x i32] [i32 1, i32 2], [2 x i32] [i32 3, i32 4]] 108; CHECK: llvm.mlir.global internal constant @nested_array_constant3(dense<[{{\[}}[1, 2], [3, 4]]]> : tensor<1x2x2xi32>) {dso_local} : !llvm.array<1 x array<2 x array<2 x i32>>> 109@nested_array_constant3 = internal constant [1 x [2 x [2 x i32]]] [[2 x [2 x i32]] [[2 x i32] [i32 1, i32 2], [2 x i32] [i32 3, i32 4]]] 110; CHECK: llvm.mlir.global internal constant @nested_array_vector(dense<[{{\[}}[1, 2], [3, 4]]]> : vector<1x2x2xi32>) {dso_local} : !llvm.array<1 x array<2 x vector<2xi32>>> 111@nested_array_vector = internal constant [1 x [2 x <2 x i32>]] [[2 x <2 x i32>] [<2 x i32> <i32 1, i32 2>, <2 x i32> <i32 3, i32 4>]] 112 113; 114; Linkage on functions. 115; 116 117; CHECK: llvm.func internal @func_internal 118define internal void @func_internal() { 119 ret void 120} 121 122; CHECK: llvm.func @fe(i32) -> f32 123declare float @fe(i32) 124 125; FIXME: function attributes. 126; CHECK-LABEL: llvm.func internal @f1(%arg0: i64) -> i32 { 127; CHECK-DAG: %[[c2:[0-9]+]] = llvm.mlir.constant(2 : i32) : i32 128; CHECK-DAG: %[[c42:[0-9]+]] = llvm.mlir.constant(42 : i32) : i32 129; CHECK-DAG: %[[c1:[0-9]+]] = llvm.mlir.constant(true) : i1 130; CHECK-DAG: %[[c43:[0-9]+]] = llvm.mlir.constant(43 : i32) : i32 131define internal dso_local i32 @f1(i64 %a) norecurse { 132entry: 133; CHECK: %{{[0-9]+}} = llvm.inttoptr %arg0 : i64 to !llvm.ptr<i64> 134 %aa = inttoptr i64 %a to i64* 135; %[[addrof:[0-9]+]] = llvm.mlir.addressof @g2 : !llvm.ptr<f64> 136; %[[addrof2:[0-9]+]] = llvm.mlir.addressof @g2 : !llvm.ptr<f64> 137; %{{[0-9]+}} = llvm.inttoptr %arg0 : i64 to !llvm.ptr<i64> 138; %{{[0-9]+}} = llvm.ptrtoint %[[addrof2]] : !llvm.ptr<f64> to i64 139; %{{[0-9]+}} = llvm.getelementptr %[[addrof]][%3] : (!llvm.ptr<f64>, i32) -> !llvm.ptr<f64> 140 %bb = ptrtoint double* @g2 to i64 141 %cc = getelementptr double, double* @g2, i32 2 142; CHECK: %[[b:[0-9]+]] = llvm.trunc %arg0 : i64 to i32 143 %b = trunc i64 %a to i32 144; CHECK: %[[c:[0-9]+]] = llvm.call @fe(%[[b]]) : (i32) -> f32 145 %c = call float @fe(i32 %b) 146; CHECK: %[[d:[0-9]+]] = llvm.fptosi %[[c]] : f32 to i32 147 %d = fptosi float %c to i32 148; FIXME: icmp should return i1. 149; CHECK: %[[e:[0-9]+]] = llvm.icmp "ne" %[[d]], %[[c2]] : i32 150 %e = icmp ne i32 %d, 2 151; CHECK: llvm.cond_br %[[e]], ^bb1, ^bb2 152 br i1 %e, label %if.then, label %if.end 153 154; CHECK: ^bb1: 155if.then: 156; CHECK: llvm.return %[[c42]] : i32 157 ret i32 42 158 159; CHECK: ^bb2: 160if.end: 161; CHECK: %[[orcond:[0-9]+]] = llvm.or %[[e]], %[[c1]] : i1 162 %or.cond = or i1 %e, 1 163; CHECK: llvm.return %[[c43]] 164 ret i32 43 165} 166 167; Test that instructions that dominate can be out of sequential order. 168; CHECK-LABEL: llvm.func @f2(%arg0: i64) -> i64 { 169; CHECK-DAG: %[[c3:[0-9]+]] = llvm.mlir.constant(3 : i64) : i64 170define i64 @f2(i64 %a) noduplicate { 171entry: 172; CHECK: llvm.br ^bb2 173 br label %next 174 175; CHECK: ^bb1: 176end: 177; CHECK: llvm.return %1 178 ret i64 %b 179 180; CHECK: ^bb2: 181next: 182; CHECK: %1 = llvm.add %arg0, %[[c3]] : i64 183 %b = add i64 %a, 3 184; CHECK: llvm.br ^bb1 185 br label %end 186} 187 188; Test arguments/phis. 189; CHECK-LABEL: llvm.func @f2_phis(%arg0: i64) -> i64 { 190; CHECK-DAG: %[[c3:[0-9]+]] = llvm.mlir.constant(3 : i64) : i64 191define i64 @f2_phis(i64 %a) noduplicate { 192entry: 193; CHECK: llvm.br ^bb2 194 br label %next 195 196; CHECK: ^bb1(%1: i64): 197end: 198 %c = phi i64 [ %b, %next ] 199; CHECK: llvm.return %1 200 ret i64 %c 201 202; CHECK: ^bb2: 203next: 204; CHECK: %2 = llvm.add %arg0, %[[c3]] : i64 205 %b = add i64 %a, 3 206; CHECK: llvm.br ^bb1 207 br label %end 208} 209 210; CHECK-LABEL: llvm.func @f3() -> !llvm.ptr<i32> 211define i32* @f3() { 212; CHECK: %[[c:[0-9]+]] = llvm.mlir.addressof @g2 : !llvm.ptr<f64> 213; CHECK: %[[b:[0-9]+]] = llvm.bitcast %[[c]] : !llvm.ptr<f64> to !llvm.ptr<i32> 214; CHECK: llvm.return %[[b]] : !llvm.ptr<i32> 215 ret i32* bitcast (double* @g2 to i32*) 216} 217 218; CHECK-LABEL: llvm.func @f4() -> !llvm.ptr<i32> 219define i32* @f4() { 220; CHECK: %[[b:[0-9]+]] = llvm.mlir.null : !llvm.ptr<i32> 221; CHECK: llvm.return %[[b]] : !llvm.ptr<i32> 222 ret i32* bitcast (double* null to i32*) 223} 224 225; CHECK-LABEL: llvm.func @f5 226define void @f5(i32 %d) { 227; FIXME: icmp should return i1. 228; CHECK: = llvm.icmp "eq" 229 %1 = icmp eq i32 %d, 2 230; CHECK: = llvm.icmp "slt" 231 %2 = icmp slt i32 %d, 2 232; CHECK: = llvm.icmp "sle" 233 %3 = icmp sle i32 %d, 2 234; CHECK: = llvm.icmp "sgt" 235 %4 = icmp sgt i32 %d, 2 236; CHECK: = llvm.icmp "sge" 237 %5 = icmp sge i32 %d, 2 238; CHECK: = llvm.icmp "ult" 239 %6 = icmp ult i32 %d, 2 240; CHECK: = llvm.icmp "ule" 241 %7 = icmp ule i32 %d, 2 242; CHECK: = llvm.icmp "ugt" 243 %8 = icmp ugt i32 %d, 2 244 ret void 245} 246 247; CHECK-LABEL: llvm.func @f6(%arg0: !llvm.ptr<func<void (i16)>>) 248define void @f6(void (i16) *%fn) { 249; CHECK: %[[c:[0-9]+]] = llvm.mlir.constant(0 : i16) : i16 250; CHECK: llvm.call %arg0(%[[c]]) 251 call void %fn(i16 0) 252 ret void 253} 254 255; CHECK-LABEL: llvm.func @FPArithmetic(%arg0: f32, %arg1: f32, %arg2: f64, %arg3: f64) 256define void @FPArithmetic(float %a, float %b, double %c, double %d) { 257 ; CHECK: %[[a1:[0-9]+]] = llvm.mlir.constant(3.030000e+01 : f64) : f64 258 ; CHECK: %[[a2:[0-9]+]] = llvm.mlir.constant(3.030000e+01 : f32) : f32 259 ; CHECK: %[[a3:[0-9]+]] = llvm.fadd %[[a2]], %arg0 : f32 260 %1 = fadd float 0x403E4CCCC0000000, %a 261 ; CHECK: %[[a4:[0-9]+]] = llvm.fadd %arg0, %arg1 : f32 262 %2 = fadd float %a, %b 263 ; CHECK: %[[a5:[0-9]+]] = llvm.fadd %[[a1]], %arg2 : f64 264 %3 = fadd double 3.030000e+01, %c 265 ; CHECK: %[[a6:[0-9]+]] = llvm.fsub %arg0, %arg1 : f32 266 %4 = fsub float %a, %b 267 ; CHECK: %[[a7:[0-9]+]] = llvm.fsub %arg2, %arg3 : f64 268 %5 = fsub double %c, %d 269 ; CHECK: %[[a8:[0-9]+]] = llvm.fmul %arg0, %arg1 : f32 270 %6 = fmul float %a, %b 271 ; CHECK: %[[a9:[0-9]+]] = llvm.fmul %arg2, %arg3 : f64 272 %7 = fmul double %c, %d 273 ; CHECK: %[[a10:[0-9]+]] = llvm.fdiv %arg0, %arg1 : f32 274 %8 = fdiv float %a, %b 275 ; CHECK: %[[a12:[0-9]+]] = llvm.fdiv %arg2, %arg3 : f64 276 %9 = fdiv double %c, %d 277 ; CHECK: %[[a11:[0-9]+]] = llvm.frem %arg0, %arg1 : f32 278 %10 = frem float %a, %b 279 ; CHECK: %[[a13:[0-9]+]] = llvm.frem %arg2, %arg3 : f64 280 %11 = frem double %c, %d 281 ret void 282} 283 284; CHECK-LABEL: llvm.func @FPComparison(%arg0: f32, %arg1: f32) 285define void @FPComparison(float %a, float %b) { 286 ; CHECK: llvm.fcmp "_false" %arg0, %arg1 287 %1 = fcmp false float %a, %b 288 ; CHECK: llvm.fcmp "oeq" %arg0, %arg1 289 %2 = fcmp oeq float %a, %b 290 ; CHECK: llvm.fcmp "ogt" %arg0, %arg1 291 %3 = fcmp ogt float %a, %b 292 ; CHECK: llvm.fcmp "oge" %arg0, %arg1 293 %4 = fcmp oge float %a, %b 294 ; CHECK: llvm.fcmp "olt" %arg0, %arg1 295 %5 = fcmp olt float %a, %b 296 ; CHECK: llvm.fcmp "ole" %arg0, %arg1 297 %6 = fcmp ole float %a, %b 298 ; CHECK: llvm.fcmp "one" %arg0, %arg1 299 %7 = fcmp one float %a, %b 300 ; CHECK: llvm.fcmp "ord" %arg0, %arg1 301 %8 = fcmp ord float %a, %b 302 ; CHECK: llvm.fcmp "ueq" %arg0, %arg1 303 %9 = fcmp ueq float %a, %b 304 ; CHECK: llvm.fcmp "ugt" %arg0, %arg1 305 %10 = fcmp ugt float %a, %b 306 ; CHECK: llvm.fcmp "uge" %arg0, %arg1 307 %11 = fcmp uge float %a, %b 308 ; CHECK: llvm.fcmp "ult" %arg0, %arg1 309 %12 = fcmp ult float %a, %b 310 ; CHECK: llvm.fcmp "ule" %arg0, %arg1 311 %13 = fcmp ule float %a, %b 312 ; CHECK: llvm.fcmp "une" %arg0, %arg1 313 %14 = fcmp une float %a, %b 314 ; CHECK: llvm.fcmp "uno" %arg0, %arg1 315 %15 = fcmp uno float %a, %b 316 ; CHECK: llvm.fcmp "_true" %arg0, %arg1 317 %16 = fcmp true float %a, %b 318 ret void 319} 320 321; Testing rest of the floating point constant kinds. 322; CHECK-LABEL: llvm.func @FPConstant(%arg0: f16, %arg1: bf16, %arg2: f128, %arg3: f80) 323define void @FPConstant(half %a, bfloat %b, fp128 %c, x86_fp80 %d) { 324 ; CHECK-DAG: %[[C0:.+]] = llvm.mlir.constant(7.000000e+00 : f80) : f80 325 ; CHECK-DAG: %[[C1:.+]] = llvm.mlir.constant(0.000000e+00 : f128) : f128 326 ; CHECK-DAG: %[[C2:.+]] = llvm.mlir.constant(1.000000e+00 : bf16) : bf16 327 ; CHECK-DAG: %[[C3:.+]] = llvm.mlir.constant(1.000000e+00 : f16) : f16 328 329 ; CHECK: llvm.fadd %[[C3]], %arg0 : f16 330 %1 = fadd half 1.0, %a 331 ; CHECK: llvm.fadd %[[C2]], %arg1 : bf16 332 %2 = fadd bfloat 1.0, %b 333 ; CHECK: llvm.fadd %[[C1]], %arg2 : f128 334 %3 = fadd fp128 0xL00000000000000000000000000000000, %c 335 ; CHECK: llvm.fadd %[[C0]], %arg3 : f80 336 %4 = fadd x86_fp80 0xK4001E000000000000000, %d 337 ret void 338} 339 340; 341; Functions as constants. 342; 343 344; Calling the function that has not been defined yet. 345; CHECK-LABEL: @precaller 346define i32 @precaller() { 347 %1 = alloca i32 ()* 348 ; CHECK: %[[func:.*]] = llvm.mlir.addressof @callee : !llvm.ptr<func<i32 ()>> 349 ; CHECK: llvm.store %[[func]], %[[loc:.*]] 350 store i32 ()* @callee, i32 ()** %1 351 ; CHECK: %[[indir:.*]] = llvm.load %[[loc]] 352 %2 = load i32 ()*, i32 ()** %1 353 ; CHECK: llvm.call %[[indir]]() 354 %3 = call i32 %2() 355 ret i32 %3 356} 357 358define i32 @callee() { 359 ret i32 42 360} 361 362; Calling the function that has been defined. 363; CHECK-LABEL: @postcaller 364define i32 @postcaller() { 365 %1 = alloca i32 ()* 366 ; CHECK: %[[func:.*]] = llvm.mlir.addressof @callee : !llvm.ptr<func<i32 ()>> 367 ; CHECK: llvm.store %[[func]], %[[loc:.*]] 368 store i32 ()* @callee, i32 ()** %1 369 ; CHECK: %[[indir:.*]] = llvm.load %[[loc]] 370 %2 = load i32 ()*, i32 ()** %1 371 ; CHECK: llvm.call %[[indir]]() 372 %3 = call i32 %2() 373 ret i32 %3 374} 375 376@_ZTIi = external dso_local constant i8* 377@_ZTIii= external dso_local constant i8** 378declare void @foo(i8*) 379declare i8* @bar(i8*) 380declare i32 @__gxx_personality_v0(...) 381 382; CHECK-LABEL: @invokeLandingpad 383define i32 @invokeLandingpad() personality i8* bitcast (i32 (...)* @__gxx_personality_v0 to i8*) { 384 ; CHECK: %[[a1:[0-9]+]] = llvm.bitcast %{{[0-9]+}} : !llvm.ptr<ptr<ptr<i8>>> to !llvm.ptr<i8> 385 ; CHECK: %[[a3:[0-9]+]] = llvm.alloca %{{[0-9]+}} x i8 {alignment = 1 : i64} : (i32) -> !llvm.ptr<i8> 386 %1 = alloca i8 387 ; CHECK: llvm.invoke @foo(%[[a3]]) to ^bb2 unwind ^bb1 : (!llvm.ptr<i8>) -> () 388 invoke void @foo(i8* %1) to label %4 unwind label %2 389 390; CHECK: ^bb1: 391 ; CHECK: %{{[0-9]+}} = llvm.landingpad (catch %{{[0-9]+}} : !llvm.ptr<ptr<i8>>) (catch %[[a1]] : !llvm.ptr<i8>) (filter %{{[0-9]+}} : !llvm.array<1 x i8>) : !llvm.struct<(ptr<i8>, i32)> 392 %3 = landingpad { i8*, i32 } catch i8** @_ZTIi catch i8* bitcast (i8*** @_ZTIii to i8*) 393 ; FIXME: Change filter to a constant array once they are handled. 394 ; Currently, even though it parses this, LLVM module is broken 395 filter [1 x i8] [i8 1] 396 resume { i8*, i32 } %3 397 398; CHECK: ^bb2: 399 ; CHECK: llvm.return %{{[0-9]+}} : i32 400 ret i32 1 401 402; CHECK: ^bb3: 403 ; CHECK: %{{[0-9]+}} = llvm.invoke @bar(%[[a3]]) to ^bb2 unwind ^bb1 : (!llvm.ptr<i8>) -> !llvm.ptr<i8> 404 %6 = invoke i8* @bar(i8* %1) to label %4 unwind label %2 405 406; CHECK: ^bb4: 407 ; CHECK: llvm.return %{{[0-9]+}} : i32 408 ret i32 0 409} 410 411; CHECK-LABEL: @hasGCFunction 412; CHECK-SAME: garbageCollector = "statepoint-example" 413define void @hasGCFunction() gc "statepoint-example" { 414 ret void 415} 416 417;CHECK-LABEL: @useFreezeOp 418define i32 @useFreezeOp(i32 %x) { 419 ;CHECK: %{{[0-9]+}} = llvm.freeze %{{[0-9a-z]+}} : i32 420 %1 = freeze i32 %x 421 %2 = add i8 10, 10 422 ;CHECK: %{{[0-9]+}} = llvm.freeze %{{[0-9]+}} : i8 423 %3 = freeze i8 %2 424 %poison = add nsw i1 0, undef 425 ret i32 0 426} 427 428;CHECK-LABEL: @useFenceInst 429define i32 @useFenceInst() { 430 ;CHECK: llvm.fence syncscope("agent") seq_cst 431 fence syncscope("agent") seq_cst 432 ;CHECK: llvm.fence release 433 fence release 434 ;CHECK: llvm.fence seq_cst 435 fence syncscope("") seq_cst 436 ret i32 0 437} 438 439; Switch instruction 440declare void @g(i32) 441 442; CHECK-LABEL: llvm.func @simple_switch(%arg0: i32) { 443define void @simple_switch(i32 %val) { 444; CHECK: %[[C0:.+]] = llvm.mlir.constant(11 : i32) : i32 445; CHECK: %[[C1:.+]] = llvm.mlir.constant(87 : i32) : i32 446; CHECK: %[[C2:.+]] = llvm.mlir.constant(78 : i32) : i32 447; CHECK: %[[C3:.+]] = llvm.mlir.constant(94 : i32) : i32 448; CHECK: %[[C4:.+]] = llvm.mlir.constant(1 : i32) : i32 449; CHECK: llvm.switch %arg0 : i32, ^[[BB5:.+]] [ 450; CHECK: 0: ^[[BB1:.+]], 451; CHECK: 9: ^[[BB2:.+]], 452; CHECK: 994: ^[[BB3:.+]], 453; CHECK: 1154: ^[[BB4:.+]] 454; CHECK: ] 455 switch i32 %val, label %def [ 456 i32 0, label %one 457 i32 9, label %two 458 i32 994, label %three 459 i32 1154, label %four 460 ] 461 462; CHECK: ^[[BB1]]: 463; CHECK: llvm.call @g(%[[C4]]) : (i32) -> () 464; CHECK: llvm.return 465one: 466 call void @g(i32 1) 467 ret void 468; CHECK: ^[[BB2]]: 469; CHECK: llvm.call @g(%[[C3]]) : (i32) -> () 470; CHECK: llvm.return 471two: 472 call void @g(i32 94) 473 ret void 474; CHECK: ^[[BB3]]: 475; CHECK: llvm.call @g(%[[C2]]) : (i32) -> () 476; CHECK: llvm.return 477three: 478 call void @g(i32 78) 479 ret void 480; CHECK: ^[[BB4]]: 481; CHECK: llvm.call @g(%[[C1]]) : (i32) -> () 482; CHECK: llvm.return 483four: 484 call void @g(i32 87) 485 ret void 486; CHECK: ^[[BB5]]: 487; CHECK: llvm.call @g(%[[C0]]) : (i32) -> () 488; CHECK: llvm.return 489def: 490 call void @g(i32 11) 491 ret void 492} 493 494; CHECK-LABEL: llvm.func @switch_args(%arg0: i32) { 495define void @switch_args(i32 %val) { 496 ; CHECK: %[[C0:.+]] = llvm.mlir.constant(44 : i32) : i32 497 ; CHECK: %[[C1:.+]] = llvm.mlir.constant(34 : i32) : i32 498 ; CHECK: %[[C2:.+]] = llvm.mlir.constant(33 : i32) : i32 499 %pred = icmp ult i32 %val, 87 500 br i1 %pred, label %bbs, label %bb1 501 502bb1: 503 %vx = add i32 %val, 22 504 %pred2 = icmp ult i32 %val, 94 505 br i1 %pred2, label %bb2, label %bb3 506 507bb2: 508 %vx0 = add i32 %val, 23 509 br label %one 510 511bb3: 512 br label %def 513 514; CHECK: %[[V1:.+]] = llvm.add %arg0, %[[C2]] : i32 515; CHECK: %[[V2:.+]] = llvm.add %arg0, %[[C1]] : i32 516; CHECK: %[[V3:.+]] = llvm.add %arg0, %[[C0]] : i32 517; CHECK: llvm.switch %arg0 : i32, ^[[BBD:.+]](%[[V3]] : i32) [ 518; CHECK: 0: ^[[BB1:.+]](%[[V1]], %[[V2]] : i32, i32) 519; CHECK: ] 520bbs: 521 %vy = add i32 %val, 33 522 %vy0 = add i32 %val, 34 523 %vz = add i32 %val, 44 524 switch i32 %val, label %def [ 525 i32 0, label %one 526 ] 527 528; CHECK: ^[[BB1]](%[[BA0:.+]]: i32, %[[BA1:.+]]: i32): 529one: ; pred: bb2, bbs 530 %v0 = phi i32 [%vx, %bb2], [%vy, %bbs] 531 %v1 = phi i32 [%vx0, %bb2], [%vy0, %bbs] 532 ; CHECK: llvm.add %[[BA0]], %[[BA1]] : i32 533 %vf = add i32 %v0, %v1 534 call void @g(i32 %vf) 535 ret void 536 537; CHECK: ^[[BBD]](%[[BA2:.+]]: i32): 538def: ; pred: bb3, bbs 539 %v2 = phi i32 [%vx, %bb3], [%vz, %bbs] 540 ; CHECK: llvm.call @g(%[[BA2]]) 541 call void @g(i32 %v2) 542 ret void 543} 544 545; Insert/ExtractValue 546; CHECK-LABEL: llvm.func @insert_extract_value_struct 547define float @insert_extract_value_struct({{i32},{float, double}}* %p) { 548 ; CHECK: %[[C0:.+]] = llvm.mlir.constant(2.000000e+00 : f64) 549 ; CHECK: %[[VT:.+]] = llvm.load %{{.+}} 550 %t = load {{i32},{float, double}}, {{i32},{float, double}}* %p 551 ; CHECK: %[[EV:.+]] = llvm.extractvalue %[[VT]][1 : i32, 0 : i32] : 552 ; CHECK-SAME: !llvm.struct<(struct<(i32)>, struct<(f32, f64)>)> 553 %s = extractvalue {{i32},{float, double}} %t, 1, 0 554 ; CHECK: %[[IV:.+]] = llvm.insertvalue %[[C0]], %[[VT]][1 : i32, 1 : i32] : 555 ; CHECK-SAME: !llvm.struct<(struct<(i32)>, struct<(f32, f64)>)> 556 %r = insertvalue {{i32},{float, double}} %t, double 2.0, 1, 1 557 ; CHECK: llvm.store %[[IV]], %{{.+}} 558 store {{i32},{float, double}} %r, {{i32},{float, double}}* %p 559 ; CHECK: llvm.return %[[EV]] 560 ret float %s 561} 562 563; CHECK-LABEL: llvm.func @insert_extract_value_array 564define void @insert_extract_value_array([4 x [4 x i8]] %x1) { 565 ; CHECK: %[[C0:.+]] = llvm.mlir.constant(0 : i8) 566 ; CHECK: llvm.insertvalue %[[C0]], %{{.+}}[0 : i32, 0 : i32] : !llvm.array<4 x array<4 x i8>> 567 %res1 = insertvalue [4 x [4 x i8 ]] %x1, i8 0, 0, 0 568 ; CHECK: llvm.extractvalue %{{.+}}[1 : i32] : !llvm.array<4 x array<4 x i8>> 569 %res2 = extractvalue [4 x [4 x i8 ]] %x1, 1 570 ; CHECK: llvm.extractvalue %{{.+}}[0 : i32, 1 : i32] : !llvm.array<4 x array<4 x i8>> 571 %res3 = extractvalue [4 x [4 x i8 ]] %x1, 0, 1 572 ret void 573} 574 575; Shufflevector 576; CHECK-LABEL: llvm.func @shuffle_vec 577define <4 x half> @shuffle_vec(<4 x half>* %arg0, <4 x half>* %arg1) { 578 ; CHECK: %[[V0:.+]] = llvm.load %{{.+}} : !llvm.ptr<vector<4xf16>> 579 %val0 = load <4 x half>, <4 x half>* %arg0 580 ; CHECK: %[[V1:.+]] = llvm.load %{{.+}} : !llvm.ptr<vector<4xf16>> 581 %val1 = load <4 x half>, <4 x half>* %arg1 582 ; CHECK: llvm.shufflevector %[[V0]], %[[V1]] [2 : i32, 3 : i32, -1 : i32, -1 : i32] : vector<4xf16>, vector<4xf16> 583 %shuffle = shufflevector <4 x half> %val0, <4 x half> %val1, <4 x i32> <i32 2, i32 3, i32 undef, i32 undef> 584 ret <4 x half> %shuffle 585} 586