Skip to content

Commit 246ba0f

Browse files
committed
fix docs
1 parent 36c2681 commit 246ba0f

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

mlir/include/mlir/Dialect/GPU/IR/CompilationAttrs.td

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -114,12 +114,23 @@ def GPU_KernelAttr : GPU_Attr<"Kernel", "kernel"> {
114114

115115
def GPU_KernelTableAttr : GPU_Attr<"KernelTable", "kernel_table"> {
116116
let description = [{
117-
GPU attribute representing a table of kernels metadata. All the attributes
118-
in the dictionary must be of type `#gpu.kernel`.
117+
GPU attribute representing a list of `#gpu.kernel` attributes. This
118+
attribute supports searching kernels by name. All kernels in the table must
119+
have an unique name.
119120

120121
Examples:
121122
```mlir
122-
#gpu.kernel_table<[#gpu.kernel<kernel0, ...>]>
123+
// Empty table
124+
#gpu.kernel_table<>
125+
126+
// Table with a single kernel
127+
#gpu.kernel_table<[#gpu.kernel<kernel0, () -> () >]>
128+
129+
// Table with multiple kernels.
130+
#gpu.kernel_table<[
131+
#gpu.kernel<"kernel0", (i32, f32) -> (), metadata = {sgpr_count = 255}>,
132+
#gpu.kernel<"kernel1", (i32) -> ()>
133+
]>
123134
```
124135
}];
125136
let parameters = (ins

0 commit comments

Comments
 (0)