[Mlir-commits] [mlir] [MLIR][Vector]Generalize DropUnitDimFromElementwiseOps (PR #92934)

Hugo Trachino llvmlistbot at llvm.org
Thu May 23 01:38:36 PDT 2024


================
@@ -459,6 +459,26 @@ func.func @fold_unit_dims_entirely(%arg0 : vector<8xi32>,
 // CHECK-128B-LABEL: func @fold_unit_dims_entirely(
 //   CHECK-128B-NOT:   memref.collapse_shape
 
+// -----
+
+func.func @fold_unit_center_dim_scalable(%arg0 : vector<8x1x[1]xf128>,
----------------
nujaa wrote:

Well unit scalable dims should not be eliminated. but it is supported. Added a test.

https://github.com/llvm/llvm-project/pull/92934


More information about the Mlir-commits mailing list