aboutsummaryrefslogtreecommitdiff
path: root/mlir/test/Dialect/Mesh/folding.mlir
diff options
context:
space:
mode:
Diffstat (limited to 'mlir/test/Dialect/Mesh/folding.mlir')
-rw-r--r--mlir/test/Dialect/Mesh/folding.mlir22
1 files changed, 0 insertions, 22 deletions
diff --git a/mlir/test/Dialect/Mesh/folding.mlir b/mlir/test/Dialect/Mesh/folding.mlir
deleted file mode 100644
index 369f316d..0000000
--- a/mlir/test/Dialect/Mesh/folding.mlir
+++ /dev/null
@@ -1,22 +0,0 @@
-// RUN: mlir-opt -test-mesh-simplifications %s | FileCheck %s
-
-mesh.mesh @mesh0(shape = 4x?x2)
-mesh.mesh @mesh1(shape = 2x3)
-
-// CHECK-LABEL: func.func @mesh_shape_op_folding
-func.func @mesh_shape_op_folding() -> (index, index) {
- // CHECK: %[[AXIS_2_SIZE:.*]] = arith.constant 2 : index
- // CHECK: %[[AXIS_1_SIZE:.*]] = mesh.mesh_shape @mesh0 axes = [1] : index
- %0:2 = mesh.mesh_shape @mesh0 axes = [2, 1] : index, index
- // CHECK: return %[[AXIS_2_SIZE]], %[[AXIS_1_SIZE]]
- return %0#0, %0#1 : index, index
-}
-
-// CHECK-LABEL: func.func @mesh_shape_op_folding_all_axes_static_mesh
-func.func @mesh_shape_op_folding_all_axes_static_mesh() -> (index, index) {
- // CHECK: %[[AXIS_0_SIZE:.*]] = arith.constant 2 : index
- // CHECK: %[[AXIS_1_SIZE:.*]] = arith.constant 3 : index
- %0:2 = mesh.mesh_shape @mesh1 : index, index
- // CHECK: return %[[AXIS_0_SIZE]], %[[AXIS_1_SIZE]]
- return %0#0, %0#1 : index, index
-}