mlir/test/mlir-cpu-runner/x86-varargs.mlir

   1 // RUN: mlir-cpu-runner %s -e caller --entry-point-result=i32 | FileCheck %s
   2 // Varaidic argument list (va_list) and the extraction logics are ABI-specific.
   3 // REQUIRES: x86-native-target
   4 // UNSUPPORTED: system-windows
   5
   6 // Check if variadic functions can be called and the correct variadic argument
   7 // can be extracted.
   8
   9 llvm.func @caller() -> i32 {
  10   %0 = llvm.mlir.constant(3 : i32) : i32
  11   %1 = llvm.mlir.constant(2 : i32) : i32
  12   %2 = llvm.mlir.constant(1 : i32) : i32
  13   %3 = llvm.call @foo(%2, %1, %0) vararg(!llvm.func<i32 (i32, ...)>) : (i32, i32, i32) -> i32
  14   llvm.return %3 : i32
  15 }
  16
  17 // Equivalent C code:
  18 // int foo(int X, ...) {
  19 //  va_list args;
  20 //  va_start(args, X);
  21 //  int num = va_arg(args, int);
  22 //  va_end(args);
  23 //  return num;
  24 //}
  25 llvm.func @foo(%arg0: i32, ...) -> i32 {
  26   %0 = llvm.mlir.constant(8 : i64) : i64
  27   %1 = llvm.mlir.constant(2 : i32) : i32
  28   %2 = llvm.mlir.constant(0 : i64) : i64
  29   %3 = llvm.mlir.constant(0 : i64) : i64
  30   %4 = llvm.mlir.constant(8 : i32) : i32
  31   %5 = llvm.mlir.constant(3 : i32) : i32
  32   %6 = llvm.mlir.constant(0 : i64) : i64
  33   %7 = llvm.mlir.constant(0 : i64) : i64
  34   %8 = llvm.mlir.constant(41 : i32) : i32
  35   %9 = llvm.mlir.constant(0 : i32) : i32
  36   %10 = llvm.mlir.constant(0 : i64) : i64
  37   %11 = llvm.mlir.constant(0 : i64) : i64
  38   %12 = llvm.mlir.constant(1 : i32) : i32
  39   %13 = llvm.alloca %12 x !llvm.array<1 x struct<"struct.va_list", (i32, i32, ptr, ptr)>> {alignment = 8 : i64} : (i32) -> !llvm.ptr
  40   llvm.intr.vastart %13 : !llvm.ptr
  41   %15 = llvm.getelementptr %13[%11, %10, 0] : (!llvm.ptr, i64, i64) -> !llvm.ptr, !llvm.array<1 x struct<"struct.va_list", (i32, i32, ptr, ptr)>>
  42   %16 = llvm.load %15 : !llvm.ptr -> i32
  43   %17 = llvm.icmp "ult" %16, %8 : i32
  44   llvm.cond_br %17, ^bb1, ^bb2
  45 ^bb1:  // pred: ^bb0
  46   %18 = llvm.getelementptr %13[%7, %6, 3] : (!llvm.ptr, i64, i64) -> !llvm.ptr, !llvm.array<1 x struct<"struct.va_list", (i32, i32, ptr, ptr)>>
  47   %19 = llvm.load %18 : !llvm.ptr -> !llvm.ptr
  48   %20 = llvm.zext %16 : i32 to i64
  49   %21 = llvm.getelementptr %19[%20] : (!llvm.ptr, i64) -> !llvm.ptr, i8
  50   %22 = llvm.add %16, %4  : i32
  51   llvm.store %22, %15 : i32, !llvm.ptr
  52   llvm.br ^bb3(%21 : !llvm.ptr)
  53 ^bb2:  // pred: ^bb0
  54   %23 = llvm.getelementptr %13[%3, %2, 2] : (!llvm.ptr, i64, i64) -> !llvm.ptr, !llvm.array<1 x struct<"struct.va_list", (i32, i32, ptr, ptr)>>
  55   %24 = llvm.load %23 : !llvm.ptr -> !llvm.ptr
  56   %25 = llvm.getelementptr %24[%0] : (!llvm.ptr, i64) -> !llvm.ptr, i8
  57   llvm.store %25, %23 : !llvm.ptr, !llvm.ptr
  58   llvm.br ^bb3(%24 : !llvm.ptr)
  59 ^bb3(%26: !llvm.ptr):  // 2 preds: ^bb1, ^bb2
  60   %28 = llvm.load %26 : !llvm.ptr -> i32
  61   llvm.intr.vaend %13 : !llvm.ptr
  62   llvm.return %28 : i32
  63 }
  64
  65 // CHECK: 2