[Mlir-commits] [mlir] [mlir][gpu] Support Cluster of Thread Blocks in `gpu.launch_func` (PR #72871)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Fri Nov 24 02:57:06 PST 2023
================
@@ -550,12 +601,23 @@ def GPU_LaunchFuncOp :GPU_Op<"launch_func", [
/// The name of the kernel.
StringAttr getKernelName();
+ /// Returns true if cluster size is specified.
+ bool hasClusterSize() {
+ if (getClusterSizeX() && getClusterSizeY() && getClusterSizeZ())
----------------
qcolombet wrote:
Is it valid to set `X`, but not `Y` and `Z`?
I'm thinking we could check if one is set and assert that the others are set too.
https://github.com/llvm/llvm-project/pull/72871
More information about the Mlir-commits
mailing list