[Mlir-commits] [mlir] [mlir][python] Add `cluster_size` to `gpu.launch_func` python binding (PR #177811)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Sat Jan 24 17:14:56 PST 2026
github-actions[bot] wrote:
<!--LLVM CODE FORMAT COMMENT: {darker}-->
:warning: Python code formatter, darker found issues in your code. :warning:
<details>
<summary>
You can test this locally with the following command:
</summary>
``````````bash
darker --check --diff -r origin/main...HEAD mlir/python/mlir/dialects/gpu/__init__.py mlir/test/python/dialects/gpu/dialect.py
``````````
:warning:
The reproduction instructions above might return results for more than one PR
in a stack if you are using a stacked PR workflow. You can limit the results by
changing `origin/main` to the base branch/commit you want to compare against.
:warning:
</details>
<details>
<summary>
View the diff from darker here.
</summary>
``````````diff
--- python/mlir/dialects/gpu/__init__.py 2026-01-25 01:02:38.000000 +0000
+++ python/mlir/dialects/gpu/__init__.py 2026-01-25 01:14:25.649368 +0000
@@ -201,12 +201,15 @@
_convert_literal_to_constant, grid_size
)
block_size_x, block_size_y, block_size_z = map(
_convert_literal_to_constant, block_size
)
- cluster_size_x, cluster_size_y, cluster_size_z = map(
- _convert_literal_to_constant, cluster_size) if cluster_size else (None, None, None)
+ cluster_size_x, cluster_size_y, cluster_size_z = (
+ map(_convert_literal_to_constant, cluster_size)
+ if cluster_size
+ else (None, None, None)
+ )
super().__init__(
async_token,
async_dependencies,
kernel,
``````````
</details>
https://github.com/llvm/llvm-project/pull/177811
More information about the Mlir-commits
mailing list