| // RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s |
| |
| module attributes {dlti.dl_spec = #dlti.dl_spec<#dlti.dl_entry<"dlti.alloca_memory_space", 5 : ui32>>, llvm.target_triple = "amdgcn-amd-amdhsa", omp.is_target_device = true} { |
| omp.private {type = private} @_QFEj_private_i32 : i32 loc(#loc1) |
| omp.private {type = private} @_QFEi_private_i32 : i32 loc(#loc1) |
| llvm.func @test() { |
| %0 = llvm.mlir.constant(1 : i64) : i64 |
| %1 = llvm.alloca %0 x i32 {bindc_name = "j"} : (i64) -> !llvm.ptr<5> loc(#loc4) |
| %2 = llvm.addrspacecast %1 : !llvm.ptr<5> to !llvm.ptr loc(#loc4) |
| %3 = llvm.alloca %0 x i32 {bindc_name = "i"} : (i64) -> !llvm.ptr<5> loc(#loc4) |
| %4 = llvm.addrspacecast %3 : !llvm.ptr<5> to !llvm.ptr |
| %5 = llvm.mlir.addressof @_QFEarray : !llvm.ptr |
| %6 = omp.map.info var_ptr(%4 : !llvm.ptr, i32) map_clauses(implicit, exit_release_or_enter_alloc) capture(ByCopy) -> !llvm.ptr {name = "i"} loc(#loc3) |
| %7 = omp.map.info var_ptr(%2 : !llvm.ptr, i32) map_clauses(implicit, exit_release_or_enter_alloc) capture(ByCopy) -> !llvm.ptr {name = "j"} loc(#loc3) |
| %8 = omp.map.info var_ptr(%5 : !llvm.ptr, !llvm.array<16384 x i32>) map_clauses(implicit, tofrom) capture(ByRef) -> !llvm.ptr {name = "array"} loc(#loc3) |
| %9 = omp.map.info var_ptr(%4 : !llvm.ptr, i32) map_clauses(implicit, exit_release_or_enter_alloc) capture(ByCopy) -> !llvm.ptr {name = "i"} loc(#loc3) |
| omp.target kernel_type(spmd) map_entries(%6 -> %arg0, %7 -> %arg2, %8 -> %arg4, %9 -> %arg5 : !llvm.ptr, !llvm.ptr, !llvm.ptr, !llvm.ptr) { |
| %10 = llvm.mlir.constant(1 : i32) : i32 |
| %11 = llvm.mlir.constant(16384 : i32) : i32 |
| omp.teams { |
| omp.distribute private(@_QFEi_private_i32 %arg5 -> %arg6 : !llvm.ptr) { |
| omp.loop_nest (%arg7) : i32 = (%10) to (%11) inclusive step (%10) { |
| omp.parallel { |
| omp.wsloop private(@_QFEj_private_i32 %arg2 -> %arg8 : !llvm.ptr) { |
| omp.loop_nest (%arg9) : i32 = (%10) to (%11) inclusive step (%10) { |
| llvm.store %arg9, %arg8 : i32, !llvm.ptr loc(#loc9) |
| omp.yield |
| } loc(#loc9) |
| } loc(#loc9) |
| omp.terminator loc(#loc9) |
| } loc(#loc9) |
| omp.yield loc(#loc9) |
| } loc(#loc9) |
| } loc(#loc9) |
| omp.terminator loc(#loc9) |
| } {omp.combined} loc(#loc9) |
| omp.terminator loc(#loc9) |
| } {omp.combined} loc(#loc15) |
| llvm.return loc(#loc9) |
| } loc(#loc14) |
| llvm.mlir.global internal @_QFEarray() {addr_space = 0 : i32} : !llvm.array<16384 x i32> { |
| %0 = llvm.mlir.zero : !llvm.array<16384 x i32> |
| llvm.return %0 : !llvm.array<16384 x i32> |
| } loc(#loc2) |
| } |
| #di_file = #llvm.di_file<"test.f90" in ""> |
| #di_null_type = #llvm.di_null_type |
| #loc1 = loc("test.f90":4:23) |
| #loc2 = loc("test.f90":4:15) |
| #loc3 = loc("test.f90":1:7) |
| #loc4 = loc("test.f90":4:18) |
| #loc9 = loc("test.f90":13:11) |
| #di_compile_unit = #llvm.di_compile_unit<id = distinct[0]<>, sourceLanguage = DW_LANG_Fortran95, file = #di_file, producer = "flang", isOptimized = true, emissionKind = LineTablesOnly> |
| #di_subroutine_type = #llvm.di_subroutine_type<callingConvention = DW_CC_program, types = #di_null_type> |
| #di_subprogram = #llvm.di_subprogram<id = distinct[1]<>, compileUnit = #di_compile_unit, scope = #di_file, name = "main", file = #di_file, subprogramFlags = "Definition|Optimized|MainSubprogram", type = #di_subroutine_type> |
| #di_subprogram1 = #llvm.di_subprogram<compileUnit = #di_compile_unit, name = "target", file = #di_file, subprogramFlags = "Definition", type = #di_subroutine_type> |
| #loc14 = loc(fused<#di_subprogram>[#loc3]) |
| #loc15 = loc(fused<#di_subprogram1>[#loc9]) |
| |
| |
| // CHECK: call void @__kmpc_distribute_static{{.*}}!dbg |
| |
