[Mlir-commits] [mlir] [mlir] Fix GPU_Launch op getPrivateAttributions function (PR #178369)

[Mehdi Amini]

================
@@ -0,0 +1,13 @@
+// RUN: mlir-opt %s -test-ir-visitors | FileCheck %s
----------------
joker-eph wrote:

You don't need to use the test-ir-visitors actually.

The test case should be:

```
module {
  func.func @main(%arg0: tensor<4x4xf32>, %arg1: tensor<*xf32>, %arg2: tensor<*xf32>, %arg3: tensor<2x3xi32>, %arg4: tensor<2x3xi32>) {
    %0 = arith.constant 1 : index
    "gpu.launch"(%0, %0, %0, %0, %0, %0) <{operandSegmentSizes = array<i32: 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0>}> ({
    }) {workgroup_attributions = 0 : i64} : (index, index, index, index, index, index) -> ()
    return
  }
}
```


Which then can be added to either:
../mlir/test/Dialect/GPU/invalid.mlir
../mlir/test/Dialect/GPU/ops.mlir

depending on whether it is valid IR or there is a diagnostic emitted (which I expect).

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