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 ; CHECK: %{{.+}} = llvm.fneg %{{.+}} : f32 282 %12 = fneg float %a 283 ; CHECK: %{{.+}} = llvm.fneg %{{.+}} : f64 284 %13 = fneg double %c 285 ret void 286} 287 288; CHECK-LABEL: llvm.func @FPComparison(%arg0: f32, %arg1: f32) 289define void @FPComparison(float %a, float %b) { 290 ; CHECK: llvm.fcmp "_false" %arg0, %arg1 291 %1 = fcmp false float %a, %b 292 ; CHECK: llvm.fcmp "oeq" %arg0, %arg1 293 %2 = fcmp oeq float %a, %b 294 ; CHECK: llvm.fcmp "ogt" %arg0, %arg1 295 %3 = fcmp ogt float %a, %b 296 ; CHECK: llvm.fcmp "oge" %arg0, %arg1 297 %4 = fcmp oge float %a, %b 298 ; CHECK: llvm.fcmp "olt" %arg0, %arg1 299 %5 = fcmp olt float %a, %b 300 ; CHECK: llvm.fcmp "ole" %arg0, %arg1 301 %6 = fcmp ole float %a, %b 302 ; CHECK: llvm.fcmp "one" %arg0, %arg1 303 %7 = fcmp one float %a, %b 304 ; CHECK: llvm.fcmp "ord" %arg0, %arg1 305 %8 = fcmp ord float %a, %b 306 ; CHECK: llvm.fcmp "ueq" %arg0, %arg1 307 %9 = fcmp ueq float %a, %b 308 ; CHECK: llvm.fcmp "ugt" %arg0, %arg1 309 %10 = fcmp ugt float %a, %b 310 ; CHECK: llvm.fcmp "uge" %arg0, %arg1 311 %11 = fcmp uge float %a, %b 312 ; CHECK: llvm.fcmp "ult" %arg0, %arg1 313 %12 = fcmp ult float %a, %b 314 ; CHECK: llvm.fcmp "ule" %arg0, %arg1 315 %13 = fcmp ule float %a, %b 316 ; CHECK: llvm.fcmp "une" %arg0, %arg1 317 %14 = fcmp une float %a, %b 318 ; CHECK: llvm.fcmp "uno" %arg0, %arg1 319 %15 = fcmp uno float %a, %b 320 ; CHECK: llvm.fcmp "_true" %arg0, %arg1 321 %16 = fcmp true float %a, %b 322 ret void 323} 324 325; Testing rest of the floating point constant kinds. 326; CHECK-LABEL: llvm.func @FPConstant(%arg0: f16, %arg1: bf16, %arg2: f128, %arg3: f80) 327define void @FPConstant(half %a, bfloat %b, fp128 %c, x86_fp80 %d) { 328 ; CHECK-DAG: %[[C0:.+]] = llvm.mlir.constant(7.000000e+00 : f80) : f80 329 ; CHECK-DAG: %[[C1:.+]] = llvm.mlir.constant(0.000000e+00 : f128) : f128 330 ; CHECK-DAG: %[[C2:.+]] = llvm.mlir.constant(1.000000e+00 : bf16) : bf16 331 ; CHECK-DAG: %[[C3:.+]] = llvm.mlir.constant(1.000000e+00 : f16) : f16 332 333 ; CHECK: llvm.fadd %[[C3]], %arg0 : f16 334 %1 = fadd half 1.0, %a 335 ; CHECK: llvm.fadd %[[C2]], %arg1 : bf16 336 %2 = fadd bfloat 1.0, %b 337 ; CHECK: llvm.fadd %[[C1]], %arg2 : f128 338 %3 = fadd fp128 0xL00000000000000000000000000000000, %c 339 ; CHECK: llvm.fadd %[[C0]], %arg3 : f80 340 %4 = fadd x86_fp80 0xK4001E000000000000000, %d 341 ret void 342} 343 344; 345; Functions as constants. 346; 347 348; Calling the function that has not been defined yet. 349; CHECK-LABEL: @precaller 350define i32 @precaller() { 351 %1 = alloca i32 ()* 352 ; CHECK: %[[func:.*]] = llvm.mlir.addressof @callee : !llvm.ptr<func<i32 ()>> 353 ; CHECK: llvm.store %[[func]], %[[loc:.*]] 354 store i32 ()* @callee, i32 ()** %1 355 ; CHECK: %[[indir:.*]] = llvm.load %[[loc]] 356 %2 = load i32 ()*, i32 ()** %1 357 ; CHECK: llvm.call %[[indir]]() 358 %3 = call i32 %2() 359 ret i32 %3 360} 361 362define i32 @callee() { 363 ret i32 42 364} 365 366; Calling the function that has been defined. 367; CHECK-LABEL: @postcaller 368define i32 @postcaller() { 369 %1 = alloca i32 ()* 370 ; CHECK: %[[func:.*]] = llvm.mlir.addressof @callee : !llvm.ptr<func<i32 ()>> 371 ; CHECK: llvm.store %[[func]], %[[loc:.*]] 372 store i32 ()* @callee, i32 ()** %1 373 ; CHECK: %[[indir:.*]] = llvm.load %[[loc]] 374 %2 = load i32 ()*, i32 ()** %1 375 ; CHECK: llvm.call %[[indir]]() 376 %3 = call i32 %2() 377 ret i32 %3 378} 379 380@_ZTIi = external dso_local constant i8* 381@_ZTIii= external dso_local constant i8** 382declare void @foo(i8*) 383declare i8* @bar(i8*) 384declare i32 @__gxx_personality_v0(...) 385 386; CHECK-LABEL: @invokeLandingpad 387define i32 @invokeLandingpad() personality i8* bitcast (i32 (...)* @__gxx_personality_v0 to i8*) { 388 ; CHECK: %[[a1:[0-9]+]] = llvm.bitcast %{{[0-9]+}} : !llvm.ptr<ptr<ptr<i8>>> to !llvm.ptr<i8> 389 ; CHECK: %[[a3:[0-9]+]] = llvm.alloca %{{[0-9]+}} x i8 {alignment = 1 : i64} : (i32) -> !llvm.ptr<i8> 390 %1 = alloca i8 391 ; CHECK: llvm.invoke @foo(%[[a3]]) to ^bb2 unwind ^bb1 : (!llvm.ptr<i8>) -> () 392 invoke void @foo(i8* %1) to label %4 unwind label %2 393 394; CHECK: ^bb1: 395 ; 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)> 396 %3 = landingpad { i8*, i32 } catch i8** @_ZTIi catch i8* bitcast (i8*** @_ZTIii to i8*) 397 ; FIXME: Change filter to a constant array once they are handled. 398 ; Currently, even though it parses this, LLVM module is broken 399 filter [1 x i8] [i8 1] 400 resume { i8*, i32 } %3 401 402; CHECK: ^bb2: 403 ; CHECK: llvm.return %{{[0-9]+}} : i32 404 ret i32 1 405 406; CHECK: ^bb3: 407 ; CHECK: %{{[0-9]+}} = llvm.invoke @bar(%[[a3]]) to ^bb2 unwind ^bb1 : (!llvm.ptr<i8>) -> !llvm.ptr<i8> 408 %6 = invoke i8* @bar(i8* %1) to label %4 unwind label %2 409 410; CHECK: ^bb4: 411 ; CHECK: llvm.return %{{[0-9]+}} : i32 412 ret i32 0 413} 414 415; CHECK-LABEL: @hasGCFunction 416; CHECK-SAME: garbageCollector = "statepoint-example" 417define void @hasGCFunction() gc "statepoint-example" { 418 ret void 419} 420 421;CHECK-LABEL: @useFreezeOp 422define i32 @useFreezeOp(i32 %x) { 423 ;CHECK: %{{[0-9]+}} = llvm.freeze %{{[0-9a-z]+}} : i32 424 %1 = freeze i32 %x 425 %2 = add i8 10, 10 426 ;CHECK: %{{[0-9]+}} = llvm.freeze %{{[0-9]+}} : i8 427 %3 = freeze i8 %2 428 %poison = add nsw i1 0, undef 429 ret i32 0 430} 431 432;CHECK-LABEL: @useFenceInst 433define i32 @useFenceInst() { 434 ;CHECK: llvm.fence syncscope("agent") seq_cst 435 fence syncscope("agent") seq_cst 436 ;CHECK: llvm.fence release 437 fence release 438 ;CHECK: llvm.fence seq_cst 439 fence syncscope("") seq_cst 440 ret i32 0 441} 442 443; Switch instruction 444declare void @g(i32) 445 446; CHECK-LABEL: llvm.func @simple_switch(%arg0: i32) { 447define void @simple_switch(i32 %val) { 448; CHECK: %[[C0:.+]] = llvm.mlir.constant(11 : i32) : i32 449; CHECK: %[[C1:.+]] = llvm.mlir.constant(87 : i32) : i32 450; CHECK: %[[C2:.+]] = llvm.mlir.constant(78 : i32) : i32 451; CHECK: %[[C3:.+]] = llvm.mlir.constant(94 : i32) : i32 452; CHECK: %[[C4:.+]] = llvm.mlir.constant(1 : i32) : i32 453; CHECK: llvm.switch %arg0 : i32, ^[[BB5:.+]] [ 454; CHECK: 0: ^[[BB1:.+]], 455; CHECK: 9: ^[[BB2:.+]], 456; CHECK: 994: ^[[BB3:.+]], 457; CHECK: 1154: ^[[BB4:.+]] 458; CHECK: ] 459 switch i32 %val, label %def [ 460 i32 0, label %one 461 i32 9, label %two 462 i32 994, label %three 463 i32 1154, label %four 464 ] 465 466; CHECK: ^[[BB1]]: 467; CHECK: llvm.call @g(%[[C4]]) : (i32) -> () 468; CHECK: llvm.return 469one: 470 call void @g(i32 1) 471 ret void 472; CHECK: ^[[BB2]]: 473; CHECK: llvm.call @g(%[[C3]]) : (i32) -> () 474; CHECK: llvm.return 475two: 476 call void @g(i32 94) 477 ret void 478; CHECK: ^[[BB3]]: 479; CHECK: llvm.call @g(%[[C2]]) : (i32) -> () 480; CHECK: llvm.return 481three: 482 call void @g(i32 78) 483 ret void 484; CHECK: ^[[BB4]]: 485; CHECK: llvm.call @g(%[[C1]]) : (i32) -> () 486; CHECK: llvm.return 487four: 488 call void @g(i32 87) 489 ret void 490; CHECK: ^[[BB5]]: 491; CHECK: llvm.call @g(%[[C0]]) : (i32) -> () 492; CHECK: llvm.return 493def: 494 call void @g(i32 11) 495 ret void 496} 497 498; CHECK-LABEL: llvm.func @switch_args(%arg0: i32) { 499define void @switch_args(i32 %val) { 500 ; CHECK: %[[C0:.+]] = llvm.mlir.constant(44 : i32) : i32 501 ; CHECK: %[[C1:.+]] = llvm.mlir.constant(34 : i32) : i32 502 ; CHECK: %[[C2:.+]] = llvm.mlir.constant(33 : i32) : i32 503 %pred = icmp ult i32 %val, 87 504 br i1 %pred, label %bbs, label %bb1 505 506bb1: 507 %vx = add i32 %val, 22 508 %pred2 = icmp ult i32 %val, 94 509 br i1 %pred2, label %bb2, label %bb3 510 511bb2: 512 %vx0 = add i32 %val, 23 513 br label %one 514 515bb3: 516 br label %def 517 518; CHECK: %[[V1:.+]] = llvm.add %arg0, %[[C2]] : i32 519; CHECK: %[[V2:.+]] = llvm.add %arg0, %[[C1]] : i32 520; CHECK: %[[V3:.+]] = llvm.add %arg0, %[[C0]] : i32 521; CHECK: llvm.switch %arg0 : i32, ^[[BBD:.+]](%[[V3]] : i32) [ 522; CHECK: 0: ^[[BB1:.+]](%[[V1]], %[[V2]] : i32, i32) 523; CHECK: ] 524bbs: 525 %vy = add i32 %val, 33 526 %vy0 = add i32 %val, 34 527 %vz = add i32 %val, 44 528 switch i32 %val, label %def [ 529 i32 0, label %one 530 ] 531 532; CHECK: ^[[BB1]](%[[BA0:.+]]: i32, %[[BA1:.+]]: i32): 533one: ; pred: bb2, bbs 534 %v0 = phi i32 [%vx, %bb2], [%vy, %bbs] 535 %v1 = phi i32 [%vx0, %bb2], [%vy0, %bbs] 536 ; CHECK: llvm.add %[[BA0]], %[[BA1]] : i32 537 %vf = add i32 %v0, %v1 538 call void @g(i32 %vf) 539 ret void 540 541; CHECK: ^[[BBD]](%[[BA2:.+]]: i32): 542def: ; pred: bb3, bbs 543 %v2 = phi i32 [%vx, %bb3], [%vz, %bbs] 544 ; CHECK: llvm.call @g(%[[BA2]]) 545 call void @g(i32 %v2) 546 ret void 547} 548 549; Insert/ExtractValue 550; CHECK-LABEL: llvm.func @insert_extract_value_struct 551define float @insert_extract_value_struct({{i32},{float, double}}* %p) { 552 ; CHECK: %[[C0:.+]] = llvm.mlir.constant(2.000000e+00 : f64) 553 ; CHECK: %[[VT:.+]] = llvm.load %{{.+}} 554 %t = load {{i32},{float, double}}, {{i32},{float, double}}* %p 555 ; CHECK: %[[EV:.+]] = llvm.extractvalue %[[VT]][1 : i32, 0 : i32] : 556 ; CHECK-SAME: !llvm.struct<(struct<(i32)>, struct<(f32, f64)>)> 557 %s = extractvalue {{i32},{float, double}} %t, 1, 0 558 ; CHECK: %[[IV:.+]] = llvm.insertvalue %[[C0]], %[[VT]][1 : i32, 1 : i32] : 559 ; CHECK-SAME: !llvm.struct<(struct<(i32)>, struct<(f32, f64)>)> 560 %r = insertvalue {{i32},{float, double}} %t, double 2.0, 1, 1 561 ; CHECK: llvm.store %[[IV]], %{{.+}} 562 store {{i32},{float, double}} %r, {{i32},{float, double}}* %p 563 ; CHECK: llvm.return %[[EV]] 564 ret float %s 565} 566 567; CHECK-LABEL: llvm.func @insert_extract_value_array 568define void @insert_extract_value_array([4 x [4 x i8]] %x1) { 569 ; CHECK: %[[C0:.+]] = llvm.mlir.constant(0 : i8) 570 ; CHECK: llvm.insertvalue %[[C0]], %{{.+}}[0 : i32, 0 : i32] : !llvm.array<4 x array<4 x i8>> 571 %res1 = insertvalue [4 x [4 x i8 ]] %x1, i8 0, 0, 0 572 ; CHECK: llvm.extractvalue %{{.+}}[1 : i32] : !llvm.array<4 x array<4 x i8>> 573 %res2 = extractvalue [4 x [4 x i8 ]] %x1, 1 574 ; CHECK: llvm.extractvalue %{{.+}}[0 : i32, 1 : i32] : !llvm.array<4 x array<4 x i8>> 575 %res3 = extractvalue [4 x [4 x i8 ]] %x1, 0, 1 576 ret void 577} 578 579; Shufflevector 580; CHECK-LABEL: llvm.func @shuffle_vec 581define <4 x half> @shuffle_vec(<4 x half>* %arg0, <4 x half>* %arg1) { 582 ; CHECK: %[[V0:.+]] = llvm.load %{{.+}} : !llvm.ptr<vector<4xf16>> 583 %val0 = load <4 x half>, <4 x half>* %arg0 584 ; CHECK: %[[V1:.+]] = llvm.load %{{.+}} : !llvm.ptr<vector<4xf16>> 585 %val1 = load <4 x half>, <4 x half>* %arg1 586 ; CHECK: llvm.shufflevector %[[V0]], %[[V1]] [2 : i32, 3 : i32, -1 : i32, -1 : i32] : vector<4xf16>, vector<4xf16> 587 %shuffle = shufflevector <4 x half> %val0, <4 x half> %val1, <4 x i32> <i32 2, i32 3, i32 undef, i32 undef> 588 ret <4 x half> %shuffle 589} 590 591; ExtractElement 592; CHECK-LABEL: llvm.func @extract_element 593define half @extract_element(<4 x half>* %vec, i32 %idx) { 594 ; CHECK: %[[V0:.+]] = llvm.load %{{.+}} : !llvm.ptr<vector<4xf16>> 595 %val0 = load <4 x half>, <4 x half>* %vec 596 ; CHECK: %[[V1:.+]] = llvm.extractelement %[[V0]][%{{.+}} : i32] : vector<4xf16> 597 %r = extractelement <4 x half> %val0, i32 %idx 598 ; CHECK: llvm.return %[[V1]] 599 ret half %r 600} 601 602; InsertElement 603; CHECK-LABEL: llvm.func @insert_element 604define <4 x half> @insert_element(<4 x half>* %vec, half %v, i32 %idx) { 605 ; CHECK: %[[V0:.+]] = llvm.load %{{.+}} : !llvm.ptr<vector<4xf16>> 606 %val0 = load <4 x half>, <4 x half>* %vec 607 ; CHECK: %[[V1:.+]] = llvm.insertelement %{{.+}}, %[[V0]][%{{.+}} : i32] : vector<4xf16> 608 %r = insertelement <4 x half> %val0, half %v, i32 %idx 609 ; CHECK: llvm.return %[[V1]] 610 ret <4 x half> %r 611} 612 613; Select 614; CHECK-LABEL: llvm.func @select_inst 615define void @select_inst(i32 %arg0, i32 %arg1, i1 %pred) { 616 ; CHECK: %{{.+}} = llvm.select %{{.+}}, %{{.+}}, %{{.+}} : i1, i32 617 %1 = select i1 %pred, i32 %arg0, i32 %arg1 618 ret void 619} 620 621; Unreachable 622; CHECK-LABEL: llvm.func @unreachable_inst 623define void @unreachable_inst() { 624 ; CHECK: llvm.unreachable 625 unreachable 626} 627