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 `[]`.
18 lines
808 B
MLIR
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?
|
|
}
|