llvm/mlir/test/Dialect/LLVMIR/inline-byval-huge.mlir

// RUN: mlir-opt -inline --split-input-file %s | FileCheck %s

// CHECK-LABEL: @byval_2_000_000_000
llvm.func @byval_2_000_000_000(%ptr : !llvm.ptr) {
  // CHECK: %[[SIZE:.+]] = llvm.mlir.constant(2000000000 : i64)
  // CHECK: "llvm.intr.memcpy"(%{{.*}}, %{{.*}}, %[[SIZE]]

  llvm.call @with_byval_arg(%ptr) : (!llvm.ptr) -> ()
  llvm.return
}

llvm.func @with_byval_arg(%ptr : !llvm.ptr { llvm.byval = !llvm.array<1000 x array<1000 x array <500 x i32>>> }) {
  llvm.return
}

// -----

// This exceeds representational capacity of 32-bit unsigned value.

// CHECK-LABEL: @byval_8_000_000_000
llvm.func @byval_8_000_000_000(%ptr : !llvm.ptr) {
  // CHECK: %[[SIZE:.+]] = llvm.mlir.constant(8000000000 : i64)
  // CHECK: "llvm.intr.memcpy"(%{{.*}}, %{{.*}}, %[[SIZE]]

  llvm.call @with_byval_arg(%ptr) : (!llvm.ptr) -> ()
  llvm.return
}

llvm.func @with_byval_arg(%ptr : !llvm.ptr { llvm.byval = !llvm.array<2000 x array<2000 x array <500 x i32>>> }) {
  llvm.return
}