xref: /llvm-project/mlir/test/Target/LLVMIR/openmp-reduction-call.mlir (revision d84252e064b3f35aa879c10e207f77e931f351d9)
127534d69STom Eccles// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s
227534d69STom Eccles
327534d69STom Eccles// Test that we don't crash when there is a call operation in the combiner
427534d69STom Eccles
5*d84252e0SSergio Afonsoomp.declare_reduction @add_f32 : f32
627534d69STom Ecclesinit {
727534d69STom Eccles^bb0(%arg: f32):
827534d69STom Eccles  %0 = llvm.mlir.constant(0.0 : f32) : f32
927534d69STom Eccles  omp.yield (%0 : f32)
1027534d69STom Eccles}
1127534d69STom Ecclescombiner {
1227534d69STom Eccles^bb1(%arg0: f32, %arg1: f32):
1327534d69STom Eccles// test this call here:
1427534d69STom Eccles  llvm.call @test_call() : () -> ()
1527534d69STom Eccles  %1 = llvm.fadd %arg0, %arg1 : f32
1627534d69STom Eccles  omp.yield (%1 : f32)
1727534d69STom Eccles}
1827534d69STom Eccles
1927534d69STom Ecclesllvm.func @simple_reduction(%lb : i64, %ub : i64, %step : i64) {
2027534d69STom Eccles  %c1 = llvm.mlir.constant(1 : i32) : i32
2127534d69STom Eccles  %0 = llvm.alloca %c1 x i32 : (i32) -> !llvm.ptr
2227534d69STom Eccles  omp.parallel reduction(@add_f32 %0 -> %prv : !llvm.ptr) {
2327534d69STom Eccles    %1 = llvm.mlir.constant(2.0 : f32) : f32
2427534d69STom Eccles    %2 = llvm.load %prv : !llvm.ptr -> f32
2527534d69STom Eccles    %3 = llvm.fadd %1, %2 : f32
2627534d69STom Eccles    llvm.store %3, %prv : f32, !llvm.ptr
2727534d69STom Eccles    omp.terminator
2827534d69STom Eccles  }
2927534d69STom Eccles  llvm.return
3027534d69STom Eccles}
3127534d69STom Eccles
3227534d69STom Ecclesllvm.func @test_call() -> ()
3327534d69STom Eccles
3427534d69STom Eccles// Call to the troublesome function will have been inlined twice: once into
3527534d69STom Eccles// main and once into the outlined function
3627534d69STom Eccles// CHECK: call void @test_call()
3727534d69STom Eccles// CHECK: call void @test_call()
38