[Mlir-commits] [mlir] [flang] [mlir][flang][openacc] Device type support on acc routine op (PR #78375)

Valentin Clement バレンタイン クレメン llvmlistbot at llvm.org
Wed Jan 17 10:59:21 PST 2024


================
@@ -2,12 +2,14 @@
 
 ! RUN: bbc -fopenacc -emit-hlfir %s -o - | FileCheck %s
 
-
+! CHECK: acc.routine @acc_routine_16 func(@_QPacc_routine18) bind("_QPacc_routine17" [#acc.device_type<host>], "_QPacc_routine16" [#acc.device_type<multicore>])
+! CHECK: acc.routine @acc_routine_15 func(@_QPacc_routine17) worker ([#acc.device_type<host>]) vector ([#acc.device_type<multicore>])
+! CHECK: acc.routine @acc_routine_14 func(@_QPacc_routine16) gang([#acc.device_type<nvidia>]) seq ([#acc.device_type<host>])
 ! CHECK: acc.routine @acc_routine_10 func(@_QPacc_routine11) seq
 ! CHECK: acc.routine @acc_routine_9 func(@_QPacc_routine10) seq
 ! CHECK: acc.routine @acc_routine_8 func(@_QPacc_routine9) bind("_QPacc_routine9a")
 ! CHECK: acc.routine @acc_routine_7 func(@_QPacc_routine8) bind("routine8_")
-! CHECK: acc.routine @acc_routine_6 func(@_QPacc_routine7) gang(dim = 1 : i32)
+! CHECK: acc.routine @acc_routine_6 func(@_QPacc_routine7) gang(1 : i64)
----------------
clementval wrote:

My only concern is that we can have multiple values here and we would just repeat the dim keyword over if we want to be consistent with the loop gang. 

```
acc.routine @acc_routine_6 func(@_QPacc_routine7) gang(dim: 1 : i64 [#acc.device_type<nvidia>], dim: 2: i64 [#acc.device_type<host>])
```

https://github.com/llvm/llvm-project/pull/78375


More information about the Mlir-commits mailing list