We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 380a694 commit 09a604cCopy full SHA for 09a604c
mlir/python/mlir/dialects/gpu/__init__.py
@@ -39,6 +39,7 @@
39
ARGUMENT_ATTRIBUTE_NAME = "arg_attrs"
40
RESULT_ATTRIBUTE_NAME = "res_attrs"
41
42
+
43
@_ods_cext.register_operation(_Dialect, replace=True)
44
class GPUFuncOp(GPUFuncOp):
45
__doc__ = GPUFuncOp.__doc__
@@ -90,8 +91,8 @@ def __init__(
90
91
self.attributes[KNOWN_BLOCK_SIZE_ATTRIBUTE_NAME] = known_block_size
92
if known_grid_size is not None:
93
if isinstance(known_grid_size, list):
- self.attributes[KNOWN_GRID_SIZE_ATTRIBUTE_NAME] = (
94
- DenseI32ArrayAttr.get(known_grid_size)
+ self.attributes[KNOWN_GRID_SIZE_ATTRIBUTE_NAME] = DenseI32ArrayAttr.get(
95
+ known_grid_size
96
)
97
else:
98
self.attributes[KNOWN_GRID_SIZE_ATTRIBUTE_NAME] = known_grid_size
0 commit comments