llvm-project/mlir/test/IR/custom-print-parse.mlir
Boian Petkantchin 944e031e36
[mlir][mesh] Use tensor shape notation for the shape of a cluster (#73826)
Examle:

substitute
mesh.cluster @mesh0(rank = 2, dim_sizes = [0, 4])

with
mesh.cluster @mesh0(rank = 2, dim_sizes = ?x4)

Same as tensor/memref shapes. The only difference is for 0-rank shapes.
With tensors you would have something like `tensor<f32>`. Here to avoid
matching an empty string a 0-rank shape is denoted by `[]`.
2023-12-08 11:34:44 -08:00

18 lines
808 B
MLIR

// RUN: mlir-opt %s | FileCheck %s
// CHECK-LABEL: module @dimension_list
module @dimension_list {
// CHECK: test.custom_dimension_list_attr dimension_list = []
test.custom_dimension_list_attr dimension_list = []
// CHECK: test.custom_dimension_list_attr dimension_list = 3
test.custom_dimension_list_attr dimension_list = 3
// CHECK: test.custom_dimension_list_attr dimension_list = 0
test.custom_dimension_list_attr dimension_list = 0
// CHECK: test.custom_dimension_list_attr dimension_list = 1x2
test.custom_dimension_list_attr dimension_list = 1x2
// CHECK: test.custom_dimension_list_attr dimension_list = ?
test.custom_dimension_list_attr dimension_list = ?
// CHECK: test.custom_dimension_list_attr dimension_list = ?x?
test.custom_dimension_list_attr dimension_list = ?x?
}