Skip to content

Commit 09a604c

Browse files
format
1 parent 380a694 commit 09a604c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

mlir/python/mlir/dialects/gpu/__init__.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
ARGUMENT_ATTRIBUTE_NAME = "arg_attrs"
4040
RESULT_ATTRIBUTE_NAME = "res_attrs"
4141

42+
4243
@_ods_cext.register_operation(_Dialect, replace=True)
4344
class GPUFuncOp(GPUFuncOp):
4445
__doc__ = GPUFuncOp.__doc__
@@ -90,8 +91,8 @@ def __init__(
9091
self.attributes[KNOWN_BLOCK_SIZE_ATTRIBUTE_NAME] = known_block_size
9192
if known_grid_size is not None:
9293
if isinstance(known_grid_size, list):
93-
self.attributes[KNOWN_GRID_SIZE_ATTRIBUTE_NAME] = (
94-
DenseI32ArrayAttr.get(known_grid_size)
94+
self.attributes[KNOWN_GRID_SIZE_ATTRIBUTE_NAME] = DenseI32ArrayAttr.get(
95+
known_grid_size
9596
)
9697
else:
9798
self.attributes[KNOWN_GRID_SIZE_ATTRIBUTE_NAME] = known_grid_size

0 commit comments

Comments
 (0)