File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
mlir/include/mlir/Dialect/XeGPU/Transforms Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -28,9 +28,8 @@ struct UnrollOptions {
2828
2929 using NativeShapeFnType =
3030 std::function<std::optional<SmallVector<int64_t >>(Operation *op)>;
31- // / Function that returns the shape of the vector to unroll to for a given
32- // / operation. The unrolling is aborted if the function returns
33- // / `std::nullopt`.
31+ // / Function that returns the shape to unroll to for a given operation.
32+ // / The unrolling is aborted if the function returns `std::nullopt`.
3433 NativeShapeFnType nativeShape = nullptr ;
3534 UnrollOptions &setNativeShapeFn (NativeShapeFnType fn) {
3635 nativeShape = std::move (fn);
@@ -59,7 +58,6 @@ void populateXeGPUSubgroupDistributePatterns(RewritePatternSet &patterns);
5958// / 4. InsertStridedSlice are inserted for VectorType result, and
6059// / BuildinUnrealizedCastOp are inserted for TensorDescType result to
6160// / re-assemble the slices into the original shape.
62- // /
6361void populateXeGPUUnrollPatterns (RewritePatternSet &patterns,
6462 const UnrollOptions &options);
6563
You can’t perform that action at this time.
0 commit comments