scalable blacklist

newling · newling · commit cd894d8c1e54 · 2025-05-06T10:29:18.000-07:00
diff --git a/mlir/lib/Dialect/Vector/Transforms/VectorLinearize.cpp b/mlir/lib/Dialect/Vector/Transforms/VectorLinearize.cpp
@@ -561,8 +561,8 @@ struct LinearizeVectorSplat final
 static bool isNotLinearizableBecauseScalable(Operation *op) {
 
   bool unsupported =
-      isa<vector::ExtractStridedSliceOp, vector::ExtractOp, vector::InsertOp>(
-          op);
+      isa<vector::ExtractStridedSliceOp, vector::InsertStridedSliceOp,
+          vector::ExtractOp, vector::InsertOp>(op);
   if (!unsupported)
     return false;
 
diff --git a/mlir/test/Dialect/Vector/linearize.mlir b/mlir/test/Dialect/Vector/linearize.mlir
@@ -215,7 +215,6 @@ func.func @insert_strided_slice_3D(%arg0 : vector<1x2x1xi8>, %arg2 : vector<3x3x
   return %0 : vector<3x3x2xi8>
 }
 
-
 // -----
 
 // CHECK-LABEL: test_vector_shuffle

Original file line number	Diff line number	Diff line change
`@@ -215,7 +215,6 @@ func.func @insert_strided_slice_3D(%arg0 : vector<1x2x1xi8>, %arg2 : vector<3x3x`
`215`	`215`	`return %0 : vector<3x3x2xi8>`
`216`	`216`	`}`
`217`	`217`
`218`		`-`
`219`	`218`	`// -----`
`220`	`219`
`221`	`220`	`// CHECK-LABEL: test_vector_shuffle`