Skip to content

Commit 4d9501a

Browse files
committed
Fixups
1 parent 89cba2d commit 4d9501a

File tree

2 files changed

+156
-143
lines changed

2 files changed

+156
-143
lines changed

mlir/lib/Dialect/ArmSME/Transforms/LegalizeForLLVMExport.cpp

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -554,13 +554,13 @@ struct VectorOuterProductToArmSMELowering
554554
///
555555
/// Example:
556556
/// ```
557-
/// %el = vector.extract %tile[%y,%x]: i32 from vector<[4]x[4]xi32>
557+
/// %el = vector.extract %tile[%row, %col]: i32 from vector<[4]x[4]xi32>
558558
/// ```
559559
/// Becomes:
560560
/// ```
561-
/// %slice = arm_sme.move_tile_slice_to_vector %tile[%y]
561+
/// %slice = arm_sme.move_tile_slice_to_vector %tile[%row]
562562
/// : vector<[4]xi32> from vector<[4]x[4]xi32>
563-
/// %el = vector.extract %slice[%x] : i32 from vector<[4]xi32>
563+
/// %el = vector.extract %slice[%col] : i32 from vector<[4]xi32>
564564
/// ```
565565
struct VectorExtractToArmSMELowering
566566
: public ConvertOpToLLVMPattern<vector::ExtractOp> {
@@ -609,14 +609,15 @@ struct VectorExtractToArmSMELowering
609609
///
610610
/// Example:
611611
/// ```
612-
/// %new_tile = vector.insert %el, %tile[%y,%x] : i32 into vector<[4]x[4]xi32>
612+
/// %new_tile = vector.insert %el, %tile[%row, %col]
613+
/// : i32 into vector<[4]x[4]xi32>
613614
/// ```
614615
/// Becomes:
615616
/// ```
616-
/// %slice = arm_sme.move_tile_slice_to_vector %tile[%y]
617+
/// %slice = arm_sme.move_tile_slice_to_vector %tile[%row]
617618
/// : vector<[4]xi32> from vector<[4]x[4]xi32>
618-
/// %new_slice = vector.insert %el, %slice[%x] : i32 into vector<[4]xi32>
619-
/// %new_tile = arm_sme.move_vector_to_tile_slice %new_slice, %tile, %y
619+
/// %new_slice = vector.insert %el, %slice[%col] : i32 into vector<[4]xi32>
620+
/// %new_tile = arm_sme.move_vector_to_tile_slice %new_slice, %tile, %row
620621
/// : vector<[4]xi32> into vector<[4]x[4]xi32>
621622
/// ```
622623
struct VectorInsertToArmSMELowering

0 commit comments

Comments
 (0)