[all-commits] [llvm/llvm-project] 944e03: [mlir][mesh] Use tensor shape notation for the sha...
Boian Petkantchin via All-commits
all-commits at lists.llvm.org
Fri Dec 8 11:34:57 PST 2023
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: 944e031e36d9515b68b320f611edfc97d5460259
https://github.com/llvm/llvm-project/commit/944e031e36d9515b68b320f611edfc97d5460259
Author: Boian Petkantchin <boian.petkantchin at amd.com>
Date: 2023-12-08 (Fri, 08 Dec 2023)
Changed paths:
M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.td
M mlir/include/mlir/IR/OpImplementation.h
M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
M mlir/lib/IR/AsmPrinter.cpp
M mlir/test/Dialect/Mesh/canonicalization.mlir
M mlir/test/Dialect/Mesh/invalid.mlir
M mlir/test/Dialect/Mesh/ops.mlir
M mlir/test/Dialect/Mesh/sharding-propagation.mlir
A mlir/test/IR/custom-print-parse.mlir
A mlir/test/IR/invalid-custom-print-parse.mlir
M mlir/test/lib/Dialect/Test/TestOps.td
Log Message:
-----------
[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 `[]`.
More information about the All-commits
mailing list